数据结构复习之n维数组实现(可变参数表的使用)

/*
    C++中好像不区分内部类与嵌套类两个名词。
    内部类与嵌套类都是指在类中定义类。
    局部类是指在函数中定义类。
    （c++不能在函数中定义函数。c++在类中定义的函数也就是成员函数。）
    （c++内部类与java内部类最大的区别就是：
    c++的内部类对象没有外部类对象的指针，不能访问外部类对象的非静态成员；
    java的非静态内部类对象有外部类对象的指针，能访问外部类对象的非静态成员。）
*/
#include<cstdarg>//标准头文件，提供宏va_start, va_arg, 和 va_end 
#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<string>
#include<cstring>
#define MAX_ARRAY_DIM 8 //假设数组的维数的最大值为8 
using namespace std;

template <typename ElemType>
class MyArray{
    private:
    public:
        class Array{
            public:
                ElemType *base;//数组元素的基址 
                int dim;//数组的维数
                int * bounds;//数组维界基址
                int * constants; //数组映像函数常量基址 
        };
         static const int ERROR = -1;
         static const int OK = 1;
        
        int InitArray(Array &A, int dim, ...);//若维数dim和随后的各维的长度合法，则构造相应的数组A 
        int DestoryArray(Array &A);//销毁数组A 
        int Value(Array A, ElemType &e, ...);//A是n为数组，e为元素变量，随后是n个下标值，将下标值对应的元素赋给e 
        int Assign(Array &A, ElemType e, ...);//将元素e赋给指定的下标值
        int Locate(Array A, va_list ap, int &off);//返回指定的下标值对应的偏移地址， 存放在off 
};

template <typename ElemType>
int MyArray<ElemType>::InitArray(Array &A, int dim, ...){
    if(dim<1 || dim > MAX_ARRAY_DIM) return MyArray::ERROR;
    A.dim = dim;
    A.bounds = (int *)malloc(dim * sizeof(int));
    if(!A.bounds) return MyArray::ERROR;
    
    int elemtotal = 1;
    va_list ap;
    va_start(ap, dim);//获取存放变长参数信息的数组 
    for(int i=0; i<dim; ++i){
        A.bounds[i] = va_arg(ap, int);
        if(A.bounds[i] < 0) return MyArray::ERROR;
        elemtotal *= A.bounds[i];
    }
    va_end(ap);
    A.base = (ElemType *)malloc(elemtotal * sizeof(ElemType));
    if(!A.base) return MyArray::ERROR;
    A.constants = (int *)malloc(dim * sizeof(int));
    if(!A.constants) return MyArray::ERROR;
    //开始求映像数组
    //int L = sizeof(ElemType);//每一个元素的大小
    int L = 1;//注意这里的元素的单位大小就是 1， 因为 A.base + off 实际上是 A.base+off*sizeof(ElemType);
    A.constants[dim-1] = L; 
    for(int i=dim-2; i>=0; --i)
        A.constants[i] = A.bounds[i+1] * A.constants[i+1];
    return MyArray::OK;
}

template <typename ElemType>
int MyArray<ElemType>::DestoryArray(Array &A){
    if(!A.base) return MyArray::ERROR;
    free(A.base);
    if(!A.bounds)  return MyArray::ERROR;
    free(A.bounds);
    if(!A.constants) return MyArray::ERROR;
    free(A.constants);
}
template <typename ElemType>
int MyArray<ElemType>::Value(Array A, ElemType &e, ...){
    va_list ap;
    int off;
    va_start(ap, e);
    if(Locate(A, ap, off)==MyArray::ERROR) return MyArray::ERROR;
    e = *(A.base+off);
    va_end(ap);
    return MyArray::OK;
}

template <typename ElemType>
int MyArray<ElemType>::Assign(Array &A, ElemType e, ...){
    va_list ap;
    int off;
    va_start(ap, e);
    if(Locate(A, ap, off)==MyArray::ERROR) return MyArray::ERROR;
    *(A.base+off) = e;
    va_end(ap);
    return MyArray::OK;
}

template <typename ElemType>
int MyArray<ElemType>::Locate(Array A, va_list ap, int &off){
    off = 0;
    for(int i=0; i<A.dim; ++i){
        int ind = va_arg(ap, int);
        if(ind<0 || ind >= A.bounds[i]) return MyArray::ERROR;
        off += A.constants[i]*ind;
    }
    return MyArray::OK;
}


class Student{
    public:
        char *name; 
        int age;
        
    Student(){
        
    }
    
    Student(char *name, int age){
        this->name = name;
        this->age = age; 
    }
    
    void outMsg(){
        cout<<"姓名: " << name << ", 年龄: " << age;
    }
};

int main(){
    MyArray<int> testArray;
    MyArray<int>::Array A;
    testArray.InitArray(A, 4, 1, 2, 3, 4);
    for(int i=0; i<1; ++i)
        for(int j=0; j<2; ++j)
            for(int k=0; k<3; ++k)
                for(int n=0; n<4; ++n){
                    int e = i+j+k+n;
                    testArray.Assign(A, e, i, j, k, n);
                }
                
    for(int i=0; i<1; ++i)
        for(int j=0; j<2; ++j)
            for(int k=0; k<3; ++k)
                for(int n=0; n<4; ++n){
                    int e;
                    testArray.Value(A, e, i, j, k, n);
                    printf("%d\n", e);
                }
                
    MyArray<Student> testArrayx;
    MyArray<Student>::Array Ax;
    testArrayx.InitArray(Ax, 3, 4, 10, 20);//4， 10， 20分别表示六层， 座位行号， 座位列号 
    Student s("hjzgg", 23);
    testArrayx.Assign(Ax, s, 1, 2, 3);
    s.name = "lxkdd"; 
    testArrayx.Value(Ax, s, 1, 2, 3);
    s.outMsg();
    return 0;
}