码迷,mamicode.com
首页 > 编程语言 > 详细

C++引用计数智能指针

时间:2015-06-20 18:27:19      阅读:149      评论:0      收藏:0      [点我收藏+]

标签:引用计数   智能指针   

#include <iostream>
using namespace std;

template<typename Type>
class auto_ptr_
{
public:
    auto_ptr_(Type *t = NULL) :ptr(t), count(new int(1))
    {
    }
    auto_ptr_(const auto_ptr_& at) :ptr(at.ptr), count(at.count)//先将count指向at的count。
    {
        count[0]++;//然后对count+1,因为多了一个对象对原对象的指向。
    }
    auto_ptr_& operator=(const auto_ptr_ &at)
    {
        if (this != &at)
        {
            if (ptr == at.ptr)return *this;//如果他们都指向同一个对象就直接返回。
            else
            {
                realse();//否则先释放他对原来对象的拥有。
                ptr = at.ptr;//重新分配一个对象。
                count = at.count;//计数指向所指对象。
                count[0]++;//计数加1。
            }
        }
        return *this;
    }
    ~auto_ptr_()
    {
        realse();
    }
    Type* operator->()
    {
        return ptr;
    }
    Type& operator*()
    {
        return *ptr;
    }
private:
    void realse()
    {
        if (--count[0] == 0)
        {
            cout << "free" << endl;
            delete []count;
            if (ptr != NULL)
            {
                delete[] ptr;
            }
        }
    }
private:
    int *count;
    Type *ptr;
};
int main()
{
    int *a = new int(1);
    auto_ptr_<int> ps(a);
    cout << *ps << endl;
    return 0;
}

C++引用计数智能指针

标签:引用计数   智能指针   

原文地址:http://blog.csdn.net/liuhuiyan_2014/article/details/46574529

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!