智能指针的简单实现

最新推荐文章于 2022-05-20 18:27:06 发布

mingxiaoshan

最新推荐文章于 2022-05-20 18:27:06 发布

阅读量216

点赞数

CC 4.0 BY-SA版权

分类专栏： c、c++ 数据结构文章标签： c++

本文链接：https://blog.youkuaiyun.com/u012979457/article/details/121195762

c、c++ 同时被 2 个专栏收录

17 篇文章

订阅专栏

数据结构

14 篇文章

订阅专栏

本文探讨了C++中智能指针的基本概念，并通过示例介绍了如何简单实现一个智能指针，以解决手动内存管理的问题。智能指针通过自动释放所指向的对象，防止内存泄漏，是C++现代内存管理的重要工具。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <iostream>
#include <assert.h>
using namespace std;


class A {
public:
    A(int a, int b) {
        x = a;
        y = b;
    };
    int x;
    int y;

    void test() {
        cout << "hello test!" << endl;
    };
};

template<typename T>
class smartPtr {
public:
    smartPtr() {
        cout << "default construct" << endl;
        _ptr = nullptr;
        _cont = nullptr;
    };

    smartPtr(T * ptr) {
        cout << "normal construct" << endl;
        _ptr = ptr;
        _cont = new int();
        (*_cont) = 1;
    };

    smartPtr(const smartPtr& sp) {
        cout << "copy construct" << endl;
        _ptr = sp._ptr;
        _cont = sp._cont;
        (*_cont)++;
    };

    smartPtr & operator=(const smartPtr& sp) {
        cout << "operator =" << endl;
        if(this->_ptr != sp._ptr) {  // 防止自赋值
            if(_ptr != nullptr) {
                (*_cont)--;
                if((*_cont) == 1) {
                    delete _ptr;
                    _ptr = nullptr;
                    delete _cont;
                    _cont = nullptr;

                }
            }
            _ptr = sp._ptr;
            _cont = sp._cont;
            (*_cont)++;
        }
        return *this;
    }

    ~smartPtr() {
        cout << "destructor" <<endl;
        if(_ptr != nullptr) {   // 指针非空时
            if((*_cont) == 1) {  // 引用计数将减为 0
                cout << "delete object" << endl;
                delete _ptr;
                _ptr = nullptr;
                delete _cont;
                _cont = nullptr;
            } else {
                (*_cont)--;  // 引用计数减小
            }
        }
    };


    T & operator*() {
        assert(_ptr != nullptr);
        return *_ptr;
    }

    T * operator->() {
        assert(_ptr != nullptr);
        return _ptr;
    }

    int getCount() {
        if(_cont != nullptr)
            return *_cont;
        return -1;
    }




private:
    T * _ptr;
    int * _cont;

};


int main()
{
    double * a = new double(10.2);
    int * b = new int(3);
    A * c = new A(1,2);
    smartPtr<double>s(a);
    smartPtr<int> s2(b);
    smartPtr<int> s3 = s2;
    smartPtr<A> s4(c);
    smartPtr<A> s5 = s4;

    s2 = s3;

    cout << *s << endl;
    cout << s.getCount() << endl;
    cout << *s2 << endl;
    cout << s.getCount() << endl;
    cout << s2.getCount() << endl;
    cout << *s3 << endl;
    cout << s.getCount() << endl;

    cout << s4->x << " " << s4->y << endl;
    s4->test();
    cout << s5->x << " " << s5->y << endl;
    cout << s5.getCount() << endl;


    return 0;
}