swap操作

Posted on | In

swap操作

我们常用的交换两个数据的操作是这样

1
2
3
4
5
6
void swap_int(int &a, int &b)
{
    int temp = a;
    a = b;
    b = temp;
}

主函数调用是这样的

1
2
3
4
int a = 100, b = 200;
swap_int(a, b);
cout << "a is " << a << endl;
cout << "b is " << b << endl;

程序输出

1
2
a is 200
b is 100

可见a和b交换了，stl为我们提供了swap函数，可以交换两个对象的数据。但大部分情况还是需要我们实现自己的swap函数，比如交换两个HasPtr对象

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

class HasPtr
{
public:
    HasPtr() = default;
    HasPtr(const string &str);
    ~HasPtr();
    HasPtr(const HasPtr &hp);
    HasPtr &operator=(const HasPtr &hp);
    friend ostream &operator<<(ostream &os, const HasPtr &);
    friend void swap(HasPtr &, HasPtr &);

private:
    string *m_str;
    int m_index;
    static int _curnum;
};

void swap(HasPtr &hptr1, HasPtr &hptr2)
{
    using std::swap;
    swap(hptr1.m_index, hptr2.m_index);
    swap(hptr1.m_str, hptr2.m_str);
}

我们在类的声明里添加了swap友元函数。然后利用了stl的swap函数完成了类内部成员的交换。交换后两个类的m_str指针指向交换了。
有了我们自己实现的swap函数，就可以简化之前的赋值运算符了。但是我们要注释掉原来版本实现的operator = ，实现新版本的

1
2
3
4
5
6
7
8
HasPtr &HasPtr::operator=(HasPtr hptr)
{
    // hptr是一个局部变量
    //交换后hptr.m_str指向this.m_str原来指向的内存
    swap(*this, hptr);
    // return返回*this后，hptr自动被回收
    return *this;
}

新版本的operator=参数为HasPtr类型，而不是引用类型，这样hptr是赋值运算符右侧变量的副本，而函数内部通过swap交换this和hptr的数据和成员，最后返回*this。随着函数返回，那么形参也就会自动销毁了，这么做的好处是代码简洁，并且不用考虑自赋值的情况。

总结

本文介绍了swap操作，合理利用swap，并为类实现swap操作，可以简化我们的操作。当我们要实现sort等排序操作，内部会用到交换逻辑，如果想实现定制化的swap就需要为类实现swap函数。
https://gitee.com/secondtonone1/cpplearn
想系统学习更多C++知识,可点击下方链接。
C++基础