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);