C&C++ 動態二維數組

1. 動態二維數組

　　目的：方便的構造一個m行n列的矩陣

 

2. 簡單實現

　　這里用函數模板寫

　　C++不支持函數模板的分離編譯，因此需要把聲明實現都寫在一個文件里

　　myarray.h：

#ifndef TESTCPLUS_MYARRAY_H
#define TESTCPLUS_MYARRAY_H

#include <iostream>
using namespace std;
template <typename T>
/*
功能：malloc申請空間部分成功，部分失敗，則回收成功的部分
*/
static void freePartMatrix(T** pary,int i)
{
    for(int j=i-1;j>=0;j--) {
        if(pary[j] != nullptr) {
            free(pary[j]);
            pary[j] = nullptr;
        }
    }
    free(pary);
}
/*
功能：獲取一個m行n列的二維數組，或者說一個m行n列的矩陣
 函數模板
*/
template <typename T>
T** getMNMatrix(int m,int n)
{
    if(m<1 || n<1) {
        return nullptr;
    }
    T** pary = nullptr;
    pary = (T**)malloc(m*sizeof(T*));
    if(pary == nullptr) {
        return nullptr;
    }
    for(int i=0;i<m;++i) {
        pary[i] = (T*)malloc(n*sizeof(T));
        if(pary[i] == nullptr) {
            freePartMatrix(pary,i);
            pary = nullptr;
            return nullptr;
        }
        memset(pary[i],0,n*sizeof(T));
    }
    return pary;
}

template <typename T>
void print_matrix(T** ary,int m,int n)
{
    if(ary == nullptr) {
        cout<<"Empty Matrix"<<endl;
    }
    for(int i=0;i<m;++i){
        for(int j=0;j<n;++j) {
            cout<<ary[i][j]<<"  ";
        }
        cout<<"\n";
    }
}
#endif //TESTCPLUS_MYARRAY_H

　　myarray.cpp

#include "myarray.h"

 

　　main.cpp

int m,n;
cin>>m>>n;
int** array = getMNMatrix<int>(m,n);
print_matrix<int>(array,m,n);