一:题目
如下为类型CMyString的声明,请为该类型添加赋值运算符函数。
以上是剑指Offer的面试题一,这里我直接实现一下整个字符串类,该面试题出现的频率还是挺高的。主要考察了类的封装、函数重载、构造函数、拷贝构造函数、深浅拷贝、赋值函数、左值、const常量、编码规范等技术点。麻雀虽小五脏俱全,知识点还挺密集的,这大概就是考察频率高的原因吧。
二:题目分析
- 构造函数形参是可缺省的,等同于函数的重载。
- 拷贝构造函数的形参是const引用,如果不使用引用,会导致在传参时,循环调用拷贝构造函数,导致栈溢出,从而出现程序崩溃的问题。
- 赋值函数返回引用类型,因为只有返回一个引用,才可以允许连续赋值。同样的形参应该为const引用,为了提高效率,防止其调用一次拷贝构造函数。
- 在赋值函数中,是否有对原有的内存进行释放,以及分配新的内存。这里需要考虑到内存泄漏的问题。
- 在赋值函数中,需要判断传入的参数是否和当前实例相同,如果相同则不进行赋值,否则在释放内存的时候,就会将传入的参数和自身的内存同时释放掉了,这个时候再进行拷贝,就会出现野指针,导致程序发生异常。
三:代码实现
EMystring.h
/*******************************
** 作者: 非正经程序员
** csdn: https://blog.csdn.net/u012058778
** 公众号: 非正经码农
** 描述: 无
**Copyright ©2021 非正经程序员版权所有
*******************************/
#ifndef E_MY_STRING_H
#define E_MY_STRING_H
class EMyString {
public:
EMyString(const char* pData = nullptr);
EMyString(const EMyString& str);
~EMyString();
EMyString& operator = (const EMyString& str);
char* cstr();
private:
char* m_pData;
};
#endif
EMystring.cpp
#include "EMyString.h"
#include <string.h>
#define OPTIMIZATION_EMSTRING
EMyString::EMyString(const char* pData)
: m_pData(nullptr) {
if (pData == nullptr) {
m_pData = new char[1];
m_pData[0] = '\0';
}
else {
int nLen = strlen(pData);
m_pData = new char[nLen + 1];
strcpy(m_pData, pData);
}
}
EMyString::EMyString(const EMyString& str)
: m_pData(nullptr) {
int nLen = strlen(str.m_pData);
m_pData = new char[nLen + 1];
strcpy(m_pData, str.m_pData);
}
EMyString::~EMyString() {
delete[] m_pData;
m_pData = nullptr;
}
char* EMyString::cstr() {
return m_pData;
}
#ifdef OPTIMIZATION_EMSTRING
EMyString& EMyString::operator =(const EMyString& str) {
if (this != &str) {
EMyString strTmp(str);
char* tempData = strTmp.m_pData;
strTmp.m_pData = m_pData; //作用域之后调用析构释放
m_pData = tempData; //tempData是在EMyString构造函数中申请的内存
}
return *this;
}
#else
EMyString& EMyString::operator = (const EMyString& str) {
if (this == &str) {
return *this;
}
delete[] m_pData;
m_pData = nullptr;
int nLen = strlen(str.m_pData);
m_pData = new char[nLen + 1];
strcpy(m_pData, str.m_pData);
return *this;
}
#endif
这里的赋值函数有一个优化的版本,就是通过利用作用域、析构函数、拷贝构造函数的特性,这里主要的优化在于如果在后面重新分配内存的时候失败了,那么我们已经把当前的内存释放掉了,导致当前的String也被破坏了。为了防止这一个现象,我们通过利用拷贝构造函数先生成一个新的字符串,然后通过交换内存,将新的内存指针赋值给当前字符串的内存指针,然后利用作用域的机制,在if之后临时实例strTmp就会调用析构函数进行释放,这时释放的内存其实就是当前字符串的内存。
如果能写出优化的版本,会给面试官留下很好的印象。
三:测试
如下是测试用例
void TestMyString() {
char pEStr[6] = { 'h', 'e', 'l', 'l', 'o', '\0'};
EMyString str(pEStr);
EMyString str1;
str1 = str;
printf("%s\n", str1.cstr());
}
测试结果如下: