智能指针

#include

using namespace std;

class U_Ptr//计数类

{

friend class HasPtr;

int *ip;

size_t use;

U_Ptr(int *p):ip(p),use(1){}

~U_Ptr(){delete ip;}

};

class HasPtr

{

public:

HasPtr(int *p,int v):ptr(new U_Ptr(p)),val(v){}//构造函数

HasPtr(const HasPtr &orig);//赋值构造函数

HasPtr& operator=(const HasPtr &orig);//操作符重载

~HasPtr();//析构函数

int *get_ptr()const {return ptr->ip;}

int get_int()const {return val;}

void set_ptr(int *p){ptr->ip=p;}

void set_int(int v){val=v;}

int get_ptr_val()const{return *ptr->ip;}

void set_ptr_val(int v){*ptr->ip=v;}

//测试用

void disp(){cout<<ptr->use<<endl;}

private:

U_Ptr *ptr;

int val;

};

HasPtr::HasPtr(const HasPtr &orig)//赋值构造函数

{

ptr=orig.ptr;

val=orig.val;

++ptr->use;

}

HasPtr& HasPtr::operator =(const HasPtr &orig)//操作符重载

{

++orig.ptr->use;//右操作数先增1

if(--ptr->use==0)//左操作数减1，并判断指向该对象的个数

delete ptr;

ptr=orig.ptr;

val=orig.val;

return *this;

}

HasPtr::~HasPtr()//析构函数

{

if(--ptr->use==0)

delete ptr;

}

int main()

{

int *obj=new int(2);//不用担心该内存的销毁，当对象ptr1、ptr2、ptr3都被销毁时，该段内存也被销毁

HasPtr ptr1(obj,42);

HasPtr ptr2(ptr1);

HasPtr ptr3=ptr1;//ptr1、ptr2、ptr3同时指向一个数据成员ptr（U_Ptr类型）

cout<<"ptr1:"<<ptr1.get_ptr_val()<<'\t'<<"ptr2:"<<ptr2.get_ptr_val()\

<<'\t'<<"ptr3:"<<ptr3.get_ptr_val()<<endl;

ptr1.disp();

ptr2.disp();

ptr3.disp();

return 0;

}