swap操作

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

void swap_int(int &a, int &b)
{
    int temp = a;
    a = b;
    b = temp;
}

主函数调用是这样的

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

程序输出

a is 200
b is 100

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


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 = ,实现新版本的

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++基础

results matching ""

    No results matching ""