智能指针share_ptr简单的写法

#include <iostream>
using namespace std;

template <class T>
class SmartPtr
{
public:
	SmartPtr(T *p);
	~SmartPtr();
	SmartPtr(const SmartPtr<T> &orig);
	SmartPtr<T>& operator=(const SmartPtr<T> &rhs);
private:
	T *ptr;
	int *use_count;
};

template<class T>
SmartPtr<T>::SmartPtr(T *p) :ptr(p)
{
	try
	{
		use_count = new int(1);
	}
	catch (...)
	{
		delete ptr;
		ptr = nullptr;
		use_count = nullptr;
		exit(1);
	}
}

template<class T>
SmartPtr<T>::~SmartPtr()
{
	if (--(*use_count)==0)
	{
		delete ptr;
		delete use_count;
		ptr = nullptr;
		use_count = nullptr;
		cout << "Destructor" << endl;
	}
}

template<class T>
SmartPtr<T>::SmartPtr(const SmartPtr<T> &orig)
{
	ptr = orig.ptr;
	use_count = orig.use_count;
	++(*use_count);
	cout << "Copy constructor" << endl;
}

template<class T>
SmartPtr<T>& SmartPtr::operator=(const SmartPtr<T> &rhs)
{
	++(*rhs.use_count);
	if (--(*use_count)==0)
	{
		delete ptr;
		delete use_count;
		cout << "Left side is delete" << endl;
	}
	ptr = rhs.ptr;
	use_count = rhs.use_count;
	cout << "Assignment operator overload" << endl;
	return *this;
}