C++ 智能指针shared_ptr的实现

转自：http://www.cnblogs.com/xuelisheng/p/9739812.html

#include <memory>
#include <iostream>
using namespace std;

template<typename T>
class smart
{
private:
	T* _ptr;
	int* _count; //reference couting

public:
	//构造函数
	smart(T* ptr = nullptr) :_ptr(ptr)
	{
		if (_ptr)
		{
			_count = new int(1);
		}
		else
		{
			_count = new int(0);
		}
	}

	//拷贝构造
	smart(const smart& ptr)
	{
		if (this != &ptr)
		{
			this->_ptr = ptr._ptr;
			this->_count = ptr._count;

			(*this->_count)++;
		}
	}

	//重载operator=
	smart operator=(const smart & ptr)
	{
		if (this->_ptr == ptr._ptr)
		{
			return *this;
		}
		if (this->_ptr)
		{
			(*this->_count)--;
			if (this->_count == 0)
			{
				delete this->_ptr;
				delete this->_count;
			}
		}
		this->_ptr = ptr._ptr;
		this->_count = ptr._count;
		(*this->_count)++;
		return *this;
	}

	//operator*重载
	T& operator*()
	{
		if (this->_ptr)
		{
			return *(this->_ptr);
		}
	}

	//operator->重载
	T& operator->()
	{
		if (this->_ptr)
		{
			return this->_ptr;
		}
	}

	//析构函数
	~smart()
	{
		(*this->_count)--;
		if (*this->_count == 0)
		{
			delete this->_ptr;
			delete this->_count;
		}
	}
	//return reference couting
	int use_count()
	{
		return *this->_count;
	}
};

int main()
{
	smart<int> sm(new int(10));
	cout << "operator*():" << *sm << endl;
	cout << "reference couting:(sm)" << sm.use_count() << endl;
	smart<int> sm2(sm);
	cout << "copy ctor reference couting:(sm)" << sm.use_count() << endl;
	smart<int> sm3;
	sm3 = sm;
	cout << "copy operator= reference couting:(sm)" << sm.use_count() << endl;
	cout << &sm << endl;
	return 0;
}

我按照原文博客把代码敲了一遍，运行结果:

转自：http://www.cnblogs.com/xuelisheng/p/9739812.html