class CMyString
{
public:
CMyString(char*pData=nullptr);
CMyString(const CMyString&str);
~CMyString(void);
private:
char*m_pData;
}
如上为类型Cmystring的声明,请为该类型添加赋值运算符函数。
class CMyString
{
public:
CMyString(char* pData = nullptr);
CMyString(const CMyString& str);
~CMyString(void);
// 添加赋值运算符函数声明
CMyString& operator=(const CMyString& str);
private:
char* m_pData;
};
CMyString& CMyString::operator=(const CMyString& str)
{
// 检查自赋值
if (this == &str)
return *this;
// 删除原有的数据
delete[] m_pData;
// 如果传入的字符串为空,则将 m_pData 设为 nullptr
if (str.m_pData == nullptr)
{
m_pData = nullptr;
}
else
{
// 分配新的内存并复制数据
size_t length = strlen(str.m_pData);
m_pData = new char[length + 1];
strcpy(m_pData, str.m_pData);
}
// 返回当前对象的引用
return *this;
}
这个实现考虑了以下几点:
- 自赋值检查:防止对象赋值给自身时出现问题。
- 删除原有数据:释放当前对象已分配的内存,防止内存泄漏。
- 处理空指针情况:如果传入的对象的 m_pData 为 nullptr,则将当前对象的 m_pData 也设为 nullptr。
- 分配新内存并复制数据:为新的字符串分配内存,并复制数据。
- 返回 *this:允许连续赋值操作,如 a = b = c。
注意:这个实现假设 CMyString 的构造函数和析构函数已正确实现,特别是析构函数应该使用 delete[] 来释放 m_pData。
你可能还想考虑实现移动赋值运算符(C++11及以后版本支持),以提高效率
CMyString& operator=(CMyString&& str) noexcept
{
if (this != &str)
{
delete[] m_pData;
m_pData = str.m_pData;
str.m_pData = nullptr;
}
return *this;
}