[C++] 동적할당을 이용한 3차원 배열

#include <iostream>
#include <cstdlib>

using namespace std;

int ***makeArray3D(int *sz);
void destroyArray3D(int ***arr, int *sz);

int ***makeArray3D(int *sz)
{
    int n1 = sz[0], n2 = sz[1], n3 = sz[2];
    int ***arr = new int **[n1];
    for(int i=0; i<n1; i++)
    {
        arr[i] = new int*[n2];
        for(int j=0; j<n2; j++)
          arr[i][j] = new int[n3];
    }
    return arr;
}

void destroyArray3D(int ***arr, int *sz)
{
  for(int i=0; i < sz[0]; i++)
  {
      for(int j=0; j < sz[1]; j++)
        delete [] arr[i][j];
      delete [] arr[i];
  }
delete [] arr;  
}

int main(int argc, char *argv[])
{
  if(argc < 4)
  {
    cout << "usage : ./str 1d 2d 3d ... nd \n";
    return -1;
  }

  int i,dim = argc-1;
  int * size = new int[dim];

  for(i = 1; i < argc; i ++) size[i-1] = atoi(argv[i]);

  int ***arr3d = NULL;

  arr3d = makeArray3D(size);
  for(int i=0; i < size[0]; i++)
    for(int j=0; j < size[1]; j++)
      for(int k=0; k< size[2]; k++)
        arr3d[i][j][k] = (i*size[1]+j)*size[2]+k;
  for(int i=0; i < size[0]; i++)
  {
    cout << "i : " << i << endl;
    for(int j=0; j<size[1]; j++)
    {
      for(int k=0; k<size[2]; k++)
        cout << arr3d[i][j][k] << ' ';
      cout << endl;
    }
    cout << endl;
  }
  destroyArray3D(arr3d, size);
  return 0;
}