当类的成员变量含有指针时,必须编写拷贝构造函数与拷贝赋值函数,否则可能发生内存泄露。
#pragma once
#ifndef STRING_H_
#define STRING_H_
#include <string.h>
#include <ostream>
#define _CRT_SECURE_NO_WARNINGS
class String
{
private:
char* m_data;
public:
char* get_c_data()const { return m_data; }
//
String(const char* str = 0)
{
if (str == nullptr)
{
m_data = new char[1];
*m_data = '\0';
}
else
{
m_data = new char[strlen(str) + 1];
strcpy_s(m_data, strlen(str) + 1, str);
}
}
//copy constructor
String(const String& str)
{
m_data = new char[strlen(str.m_data) + 1];
strcpy_s(m_data, strlen(str.m_data)+1, str.m_data);
}
String& operator=(const String& str)
{
if (&str == this) //判断是否是同一个对象
return *this;
delete[] m_data; //清空data
m_data = new char[strlen(str.m_data) + 1];
strcpy_s(m_data,strlen(str.m_data)+1, str.m_data);
return *this;
}
};
#include <iostream>
using namespace std;
ostream& operator<<(ostream& os, const String& str)
{
os << str.get_c_data();
return os;
}
#endif // !STRING_H_
在写拷贝复制构造函数时,注意需要先判断是否复制的是自身,否则程序会运行时错误。