今天看到有人問,在2維或者多維數組的情況下怎樣重載像[][]這樣的操作符號。
其實思路也不難啊,只不過是重載2個[]符號而已,并且2個[]的功能各不一樣。
第一個[]應該定位行.
最后一個[]應該返回最終數據的引用。
貼一下實現基本功能的代碼吧,像越界檢測,及其他功能就沒寫了,只要體現了思路,其他的功能好加。
using namespace std;
template <class T> class arr;
template <class T> class arrBody
{
private:
friend class arr<T>;
T* data;
int row,col,current_row;
arrBody(int r,int c,T d):row(r),col(c)
{
data=new T[r*c];
current_row=-1;
for(int k=0;k<r*c;k++) //初始化數據,默認為0
data[k]=d;
}
public:
T& operator[](int j) //重載第2個[]號
{
if(j>=0&&j<col)
return data[current_row*col+j];
}
~arrBody(){delete[]data;} }; template <class T> class arr
{
private:
arrBody<T> tBody;
public:
arrBody<T> &operator[](int i) //重載第一個[]號
{
if(i>=0&&i<tBody.row)
tBody.current_row=i;
return tBody;
}
arr(int i,int j,T d=0):tBody(i,j,d) {}
};
void main()
{
arr<int> a(10,20);
arr<double> b(5,5);
cout<<a[5][5]<<endl;
}
新聞熱點
疑難解答
圖片精選