為瞭解決上面的問題,我們應該寫一個特殊的賦值運算子函數來處理這類問題。當需要為同一個類的兩個對象相互賦值時,就可以重載運算子函數。這個方法可以解決類的賦值和指標的釋放。
下面的程式中,類中的賦值函數用new運算子從堆中分配了一個不同的指標,該指標擷取賦值對象中相應的值,然後拷貝給接受賦值的對象。
在類中重載賦值運算子的格式如下:
void operator = (const Date&) 後面我們回加以改進。目前,重載的運算子函數的傳回型別為void。它是類總的成員函數,在本程式紅,是Date類的成員函數。它的函數名始終是operator =,參數也始終是同一個類的對象的引用。參數表示的是來源物件,即賦值資料的提供者。重載函數的運算子作為目標對象的成員函數來使用。
#include "iostream.h"
#include "string.h"
class Date
{
int mo,da,yr;
char *month;
public:
Date(int m=0, int d=0, int y=0);
~Date();
void operator=(const Date&);
void display() const; };
Date::Date(int m, int d, int y)
{
static char *mos[] =
{
"January","February","March","April","May","June",
"July","August","September","October","November","December" };
mo = m; da = d; yr = y;
if (m != 0)
{ month = new char[strlen(mos[m-1])+1];
strcpy(month, mos[m-1]);
}
else month = 0;
}
Date::~Date()
{
delete [] month;
}
void Date::display()
const
{
if (month!=0) cout《month《' '《da《","《yr《endl;
}
void Date::operator=(const Date& dt)
{
if (this != &dt)
{
mo = dt.mo;
da = dt.da;
yr = dt.yr;
delete [] month;
if (dt.month != 0)
{
month = new char
[std::strlen(dt.month)+1];
std::strcpy(month, dt.month);
}
else month = 0;
}
}
int main()
{
Date birthday(8,11,1979);
birthday.display();
Date newday(12,29,2003);
newday.display();
newday = birthday;
newday.display();
return 0;
}
除了為Date類加入了一個重載運算子函數,這個程式和上面的一個程式是相同的。賦值運算子函數首先取得所需的資料,然後用delete把原來的month指標所佔用的記憶體返還給堆。接著,如果來源物件的month指標已經初始化過,就用new運算子為對象重新分配記憶體,並把來源物件的month字串拷貝給接受方。
重載的Date類賦值運算子函數的第一個語句比較了來源物件的地址和this指標。這個操作取保對象不會自己給自己賦值。