String--引用计数写时拷贝

最新推荐文章于 2022-02-13 20:47:04 发布

原创最新推荐文章于 2022-02-13 20:47:04 发布 · 361 阅读

0 ·

CC 4.0 BY-SA版权

CPP 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一个使用写时拷贝技术实现的字符串类String。该类通过引用计数来管理内存，当需要修改字符串时才进行实际拷贝，从而提高了效率。文章详细展示了String类的构造函数、析构函数、赋值操作符重载等关键成员函数的实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

写时拷贝
这里写图片描述

class String
{
public:
    String(char* str = "")
    {
        _str = new char[strlen(str) + 1];
        _refCountPtr = new int(1);
        strcpy(_str, str);  
    }
    ~String()
    {
        if (--*_refCountPtr = 0)
        {
            delete[] _str;
            delete _refCountPtr;
        }
    }
    String(String& s)
        :_str(s._str)
        , _refCountPtr(s._refCountPtr)
    {
        (*_refCountPtr)++;
    }
    String& operator = (const String& s)
    {
        if (_str != s._str)
        {
            if (--(*_refCountPtr) == 0)
            {

                delete[] _str;
                delete _refCountPtr;
            }

            _str = s._str;
            _refCountPtr = s._refCountPtr;
            (*_refCountPtr)++;
        }
        return *this;
    }
    void CopyOnWrite(char ch)
    {
        if (*_refCountPtr > 1)
        {
            char* newStr = new char[_capacity + 1];
            strcpy(newStr, _str);
            (*_refCountPtr)--;
            _str = newStr;
            _refCountPtr = new int(1);
        }
    }
    void PushBack(char ch)
    {
        CopyOnWrite(ch);
        if (_size == _capacity)
        {
            Expand(2 * _capacity);
        }
        _str[_size++] = ch;
        _str[_size] = '\0';
    }
    void Expand(size_t n)
    {
        if (n > _capacity)
        {
            _str =(char*) realloc(_str, n + 1);
            assert(_str);
            _capacity = n;
        }
    }
    char& operator[](size_t pos)
    {
        CopyOnWrite(pos);
        return _str[pos];
    }
    char operator[](size_t pos)const
    {
        return _str[pos];
    }
private:
    char* _str;
    size_t _size;
    size_t _capacity;
    int* _refCountPtr;
};