写时拷贝实现String类

写时拷贝技术是通过"引用计数"实现的，在分配空间的时候多分配4个字节，用来记录有多少个指针指向块空间，当有新的指针指向这块空间时，引用计数加一，当要释放这块空间时，引用计数减一(假装释放)，直到引用计数减为0时才真的释放掉这块空间。当有的指针要改变这块空间的值时，再为这个指针分配自己的空间(注意这时引用计数的变化，旧的空间的引用计数减一，新分配的空间引用计数加一)。

写时拷贝实现String类

代码实现：

#include <iostream>
using namespace std;

#pragma warning(disable:4996)

class String
{
public:
	//构造函数
	String(char* ptr = NULL):_str(new char[strlen(ptr) + 5])
	{
		_str = _str + 4;
		strcpy(_str, ptr);
		getCount(_str) = 1;
	}
	
	//拷贝构造函数
	String(const String& rhs):_str(rhs._str)
	{
		++getCount(_str);
	}
	String& operator=(const String& rhs)
	{
		if (this != &rhs)
		{
			Release();
			_str = rhs._str;
			++getCount(_str);
		}
		return *this;
	}

	//重载[]运算符
	char& operator[](int index)
	{
		if (getCount(_str) > 1)
		{
			--getCount(_str);
			char* tmp = new char[strlen(_str) + 5];
			tmp = tmp + 4;
			getCount(tmp) = 1;
			strcpy(tmp, _str);
			_str = tmp;
		}
		return _str[index];
	}

	//析构函数
	~String()
	{
		Release();
	}

	char* c_str()
	{
		return _str;
	}
private:
	void Release()
	{
		if (--getCount(_str) == 0)
		{
			delete[](_str - 4);
		}
	}
	int& getCount(char* ptr)
	{
		return *(int*)(ptr - 4);
	}
	char* _str;
};

int main()
{
	String s1("hello");
	String s2(s1);

	printf("s1.address:%x\n", s1.c_str());
	printf("s2.address:%x\n", s2.c_str());

	s2[0] = 's';
	printf("s1.address:%x\n", s1.c_str());
	printf("s2.address:%x\n", s2.c_str());

	s1[1] = 'w';
	printf("s1.address:%x\n", s1.c_str());
	printf("s2.address:%x\n", s2.c_str());

	return 0;
}

运行结果：