自己实现一个智能指针

要实现一个智能指针主要实现下面几个函数

1 构造函数
2 拷贝构造函数
3 析构函数
4 赋值运算符函数
5 获取引用计数函数

重点
1 构造,拷贝构造 +1
2 析构函数会使引用计数-1.
3 赋值运算符会使之前的引用计数-1。使新赋值过来的引用计数+1

#include <iostream>

template<class T>
class SmartPtr
{
public:
    SmartPtr(T* p = nullptr)
    {
        ptr = p;
        if (p == nullptr)
            ref_count = new int(0);
        else
            ref_count = new int(1);
    }
    ~SmartPtr()
    {
        if (ptr)
        {
            if (--(*ref_count) == 0)
            {
                delete ptr;
                delete ref_count;
            }
        }
    }

    SmartPtr(const SmartPtr& s)
    {
        ptr = s.ptr;
        ref_count = s.ref_count;
        (*ref_count)++;
    }

    SmartPtr& operator=(const SmartPtr& s)
    {
        if (ptr == s.ptr)
            return *this;

        if (ptr)
        {
            if (--(*ref_count) == 0)
            {
                delete ptr;
                delete ref_count;
            }
        }
        

        ptr = s.ptr;
        ref_count = s.ref_count;
        (*ref_count)++;

        return *this;
    }

    T* operator->()
    {
        return ptr;
    }

    SmartPtr& operator*()
    {
        return *this;
    }

    int use_count()
    {
        return *ref_count;
    }

private:
    T* ptr;
    int *ref_count;

};

int main()
{
	{
	    int *arry = new int[3];
	    SmartPtr<int> sp(arry);
	    SmartPtr<int> sp1;
	    sp1 = sp;
	    SmartPtr<int> sp2(sp1);
	    sp1.use_count();
	}
	std::cout << "end" << std::endl;
	return 0;
}
  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值