C++ 实现智能指针（带引用计数）-蒲公英云

C++ 实现智能指针（带引用计数）

叁歲伎倆 2022-12-31 02:23 126阅读 0赞

#include<bits/stdc++.h>
using namespace std;
//实现一个带引用计数的智能指针
/* 思路:对象有两个成员变量，一个成员变量指向资源，一个成员变量指向计数。(计数必须在堆上分配) 构造函数：让资源指针指向资源，并且创建一个和这个资源配对的int=1。 析构函数：判断int==0,决定是否释放资源 拷贝构造：int++，_p指向同一个资源 拷贝赋值运算符：检查自赋值，释放之前资源，赋值 解引用*：返回引用T 箭头：返回p指针、 */
template<typename T>
class Cnt{ 
public:
    Cnt(T*p=NULL):_p(p){ 
        if(_p!=NULL){ 
            count=1;
        }else{ 
            count=0;
        }
    }
    int getcnt(){ 
        return count;
    }
    void sub(){ 
        count--;
    }
    void add(){ 
        count++;
    }
private :
    T *_p;
    int count;    
};
template<typename T>
class ssmart{ 
public:
    ssmart(T*p=NULL):_p(p),cnt(){ //nullptr
        cnt= new Cnt<T>(_p);
    }
    ~ssmart(){ 
        if(cnt->getcnt()==1){ 
            delete cnt;
            delete _p;            
        }else{ 
            cnt->sub();
        }
    }
    ssmart(const ssmart&smart){ 
        _p=smart._p;
        cnt=smart.cnt;
        cnt->add();
    }
    ssmart& operator=(const ssmart&smart){ 
        if(this==smart)return *this;
        if(_p!=NULL){ 
            cnt->sub();
            if(!cnt->getcnt()){ 
                delete cnt;
                delete _p;                
            }
        }
        _p=smart._p;
        cnt->add();
        return *this;
    }
    T& operator*(){ 
        return *_p;
    }
    T& operator->(){ 
        return _p;
    }
private:
    T*_p;
    Cnt<T>* cnt;
};
int main(){ 
    ssmart<int> p(new int);
    *p=15;
    cout<<*p<<endl;
    ssmart<int> p1(p);
    cout<<*p1<<endl;
    return 0;
}
########################################################################################
1.29练习：
#include<bits/stdc++.h>
using  namespace std;
template<typename T>
class smart{ 
public :
    smart(T *p){ 
        _p=p;
        _cnt= new int(1);
        cout<<"ok"<<endl;
    }
    ~smart(){ 
        (*_cnt)--;
        if(*_cnt==0)delete _p;
    }
    smart(const smart&rhs){ 
        _cnt=rhs._cnt;
        (*_cnt)++;
        _p=rhs._p;
    }
    smart& operator=(const smart&rhs){ 
        if(&rhs==this)return *this;
        if(*_cnt-- == 1){ 
            delete _p;
        }
        _cnt=rhs._cnt;
        (*_cnt)++;
        _p=rhs._p;
        return *this;
    }
    int& operator*(){ 
        return *_p;
    }
    int* operator->(){ 
        return _p;
    }
    void show(){ 
        cout<<"cnt:"<< *_cnt<<" "<<"p:"<< *_p <<endl;
    }
private :
    int *_cnt;
    T *_p;
};
int main(){ 
    int *p=new int(1100) ;
    smart<int> data(p);
    *data=100;
    cout<<*data<<endl;
    data.show();
    return 0;
}