原题
LintCode 208. Assignment Operator Overloading (C++ Only)
Description
Implement an assignment operator overloading method.
Make sure that:
The new data can be copied correctly
The old data can be deleted / free correctly.
We can assign like A = B = C
Clarification
This problem is for C++ only as Java and Python don't have overloading for assignment operator.
Example
If we assign like A = B, the data in A should be deleted, and A can have a copy of data from B.
If we assign like A = B = C, both A and B should have a copy of data from C.
代码
class Solution {
public:
char *m_pData;
Solution() {
this->m_pData = NULL;
}
Solution(char *pData) {
if (pData) {
m_pData = new char[strlen(pData) + 1];
strcpy(m_pData, pData);
} else {
m_pData = NULL;
}
}
// Implement an assignment operator
Solution operator=(const Solution &object) {
// write your code here
// if (m_pData) delete[] m_pData;
if (this == &object) return *this;
if (object.m_pData) {
if (m_pData) delete[] m_pData;
m_pData = new char[strlen(object.m_pData) + 1];
strcpy(m_pData, object.m_pData);
} else {
m_pData = NULL;
}
return *this;
}
};
网友评论