重载二维数组时,调用时如a[1][2], 其实质应该是:(假设数组类型为int)
a.operator[](1)返回int*,int*作为行指针调用(int*)[2],达到目的。
分享mooc郭炜老师的一道题目:(关于三维数组下标符重载)
#include <iostream>
#include <iomanip>
#include <cstring>
using namespace std;
template <class T>
class CArray3D
{
//your code starts here
public:
T * p;
int L,r,c; //���鹲L�㣬ÿ��r��,ÿ��c��
class Aid {
public:
T * lp; // lp��ijһ���Ԫ�ص���ʼ��ַ
int c; //���鹲L�㣬ÿ��r��,ÿ��c��
Aid( T * p_,int c_):lp(p_),c(c_) { }
T * operator[]( int r ) {
return lp + r * c; //����ֵ���DZ���ĵ� r�е�Ԫ�ص���ʼ��ַ
}
operator T * () {
return lp ; //����ֵ���DZ����Ԫ�ص���ʼ��ַ
}
};
CArray3D(int l_,int r_,int c_ ):L(l_),r(r_),c(c_) {
p = new T [L * r * c];
}
Aid operator[](int L) { //���صĶ������p��¼�˵�L��Ԫ�ص���ʼ��ַ
T * lp = p + L * r * c;
return Aid(lp,c);
}
~CArray3D() { delete [] p; }
//your code ends here
};
CArray3D<int> a(3,4,5);
CArray3D<double> b(3,2,2);
void PrintA()
{
for(int i = 0;i < 3; ++i) {
cout << "layer " << i << ":" << endl;
for(int j = 0; j < 4; ++j) {
for(int k = 0; k < 5; ++k)
cout << a[i][j][k] << "," ;
cout << endl;
}
}
}
void PrintB()
{
for(int i = 0;i < 3; ++i) {
cout << "layer " << i << ":" << endl;
for(int j = 0; j < 2; ++j) {
for(int k = 0; k < 2; ++k)
cout << b[i][j][k] << "," ;
cout << endl;
}
}
}
int main()
{
int No = 0;
for( int i = 0; i < 3; ++ i ) {
a[i];
for( int j = 0; j < 4; ++j ) {
a[j][i];
for( int k = 0; k < 5; ++k )
a[i][j][k] = No ++;
a[j][i][i];
}
}
PrintA();
memset(a[1],-1 ,20*sizeof(int));
memset(a[1],-1 ,20*sizeof(int));
PrintA();
memset(a[1][1],0 ,5*sizeof(int));
PrintA();
for( int i = 0; i < 3; ++ i )
for( int j = 0; j < 2; ++j )
for( int k = 0; k < 2; ++k )
b[i][j][k] = 10.0/(i+j+k+1);
PrintB();
int n = a[0][1][2];
double f = b[0][1][1];
cout << "****" << endl;
cout << n << "," << f << endl;
return 0;
}