careercup-C和C++ 13.8

13.8 编写一个智能指针类。智能指针是一种数据类型，一般用模板实现，模拟指针行为的同时还提供自动垃圾回收机制。它会自动记录SmartPointer<T*>对象的引用计数，一旦T类型对象的引用计数为零，就会释放该对象。

解法：

智能指针跟普通指针一样，但他借助自动化内存管理保证了安全性，避免了诸如悬挂指针、内存泄漏和分配失败等问题。

智能指针必须为给定对象的所有引用维护单一引用计数。主要实现构造函数、复制构造函数和赋值运算符，析构函数。

 

C++实现代码：

#include<iostream>

#include<cstdlib>

#include<new>

using namespace std;



template <typename T>

class SmartPointer

{

public:

    SmartPointer(T* ptr)

    {

        ref=ptr;

        ref_count=(unsigned*)malloc(sizeof(unsigned));

        *ref=1;

    }

    SmartPointer(SmartPointer<T> &sptr)

    {

        ref=sptr.ref;

        ref_count=sptr.ref_count;

        ++(*ref_count);

    }

    SmartPointer<T> &operator=(SmartPointer<T> &sptr)

    {

        if(this==&sptr) return *this;

        if(*ref_count>0)

        {

            remove();

        }

        ref=sptr.ref;

        ref_count=sptr.ref_count;

        ++(*ref_count);

        return *this;

    }

    ~SmartPointer()

    {

        remove();

    }



    T getValue()

    {

        return *ref;

    }

protected:

    void remove()

    {

        --(*ref_count);

        if(*ref_count==0)

        {

            delete ref;

            free(ref_count);

            ref=NULL;

            ref_count=NULL;

        }

    }

private:

    T* ref;

    unsigned *ref_count;

};



int main()

{

    int *p1=new int();

    *p1=1111;

    int *p2=new int();

    *p2=22222;

    SmartPointer<int> sp1(p1),sp2(p2);

    SmartPointer<int> spa=sp1;

    sp2=spa;

}