작성한 질문수 6

0

원하시는 것이 아래와 같은 코드 맞나요?

/*	배열의 동적 할당/재할당/해제 함수
	2013-04-30
	KT		*/

#include <stdio.h>
#include <stdlib.h>
#include <malloc.h>


///  dim(&arr, x) : 일차원 배열의 메모리 할당
template <typename T>
int dim(T* &arr, int x)
{
	arr = (T*)malloc(sizeof(T)*x);
	return 0;
}


///  dim(arr, x, y) : 이차원 배열의 메모리 할당
template <typename T>
int dim(T** &arr, int x, int y)
{
	int i;

	arr = (T**)malloc(sizeof(T*)*x);
	for (i=0;i<x;++i)
		arr[i] = (T*)malloc(sizeof(T)*y);

	return 0;
}


///  redim(arr, x) : 일차원 배열의 메모리 재할당
template <typename T>
int redim(T* &arr, int x)
{
	arr = (T*)realloc(arr, sizeof(T)*x);
	return 0;
}


///  redim(arr, x, y) : 이차원 배열의 메모리 재할당
template <typename T>
int redim(T** &arr, int x, int y)
{
	int i;
	int s = sizeof(T);
	int ps = sizeof(T*);
	int xb = _msize(arr)/ps; // 재할당 전 행의 수

	if (xb > x)	 // 행이 줄어드는 경우 필요없는 행은 free()
	{
		for (i=x;i<xb;++i)
			free(arr[i]);
	}

	arr = (T**)realloc(arr,ps*x);

	if (xb >= x)
	{
		for (i=0;i<x;++i)
			arr[i] = (T*)realloc(arr[i],s*y);
	}
	else //행이 더 늘어나는 경우
	{
		for (i=0;i<xb;++i)
			arr[i] = (T*)realloc(arr[i],s*y);
		for (i=xb;i<x;++i)
			arr[i] = (T*)malloc(s*y);
	}

	return 0;
}


///  undim(arr) : 일차원 배열의 메모리 할당 해제
template <typename T>
void undim(T* &arr)
{
	free(arr);
	arr = NULL;
}


///  undim(arr) : 이차원 배열의 메모리 할당 해제
template <typename T>
void undim(T** &arr)
{
	int i;
	int x = _msize(arr)/sizeof(T*);	//  x = 1차원 포인터의 수 = 배열의 행 갯수

	for (i=0;i<x;i++)
	{
		free(arr[i]);
		arr[i] = NULL;
	}

	free(arr);
	arr = NULL;
}


int main()
{
	int* a;
	int** arr;


	dim(arr,20,5);
	redim(arr,5,5);
	redim(arr,7,7);


	arr[1][1] = 3;


	dim(a,100);
	redim(a,200);


	a[10] = 50;
	a[11] = 142;

	viewMemMap(arr);

	undim(arr);
	undim(a);

}