|
http://www.hackerschool.org/HS_Boards/view.php?id=books&no=1582 [복사]
책에 나와있는 소스인데요...put_heap() 함수에서
temp = (int *)malloc( count * sizeof(int) );
if( temp == NULL )
return 0;
위 소스부분에서 (int *)malloc()에서 count = 0일때 (int *)malloc(0)이 되는데
재 생각에는 이때 temp에 NULL 값이 들어가는 것 같거든요 실제로 컴파일해서
실행해보면 책에 있는 소스 대로라면 계속 위부분에서 0값이 계속 리턴 되서
실행이 되지 않습니다..멍멍님 이거 해결좀 해주세요 아래는 전체 소스구요..
/* pre_processing command */
#include <stdio.h>
#include <stdlib.h>
/* variables announce */
int *heap, count;
/* function prototypes announce */
int create_heap( void );
int put_heap( int data );
int get_heap( void );
void clean_heap( void );
/* structure define */
/* main() function define */
int main( void )
{
int n,d;
create_heap();
while( 1 ){
printf( "\n--------------------\n" );
printf( "1. Put a data\n" );
printf( "2. Get a data\n" );
printf( "3. Exit\n" );
printf( ">" );
scanf( "%d", &n );
if( n == 3 )
break;
if( n == 1 ){
printf( "Enter a number\n" );
printf( ">" );
scanf( "%d", &d );
put_heap( d );
continue;
}
if( n == 2 ){
d = get_heap();
if( d != -1 )
printf( "Number is %d.\n", d );
else
printf( "No data.\n" );
continue;
}
}
clean_heap();
return 1;
}
/* function define */
/* create_heap() define */
int create_heap( void )
{
count = 0;
heap = (int *)malloc( 1 );
if( heap == NULL ){
return 0;
}
return 1;
}
/* put_heap() define */
int put_heap( int data )
{
int *temp;
temp = (int *)malloc( count * sizeof(int) );
if( temp == NULL )
return 0;
memcpy( temp, heap, count * sizeof(int) );
free( heap );
count += 1;
heap = (int *)malloc( count * sizeof(int) );
if( heap == NULL )
return 0;
heap[0] = data;
memcpy( heap+1, temp, (count-1) * sizeof(int) );
free( temp );
return 1;
}
/* get_heap() define */
int get_heap( void )
{
int *temp,value;
if( count == 0 ){
return -1;
}
temp = (int *)malloc( count * sizeof(int) );
if( temp == NULL ){
return -1;
}
memcpy( temp, heap, count * sizeof(int) );
free( heap );
count -= 1;
heap = (int *)malloc( count * sizeof(int) );
if( heap == NULL )
return -1;
memcpy( heap, temp, count * sizeof(int) );
value = temp[count];
free( temp );
return value;
}
/* clean_heap() define */
void clean_heap( void )
{
free( heap );
}
|
Hit : 939 Date : 2003/05/18 06:07
|
2587, 
1/130 
멍멍
