include
include
include
typedef struct stHuNode{
int data;
struct stHuNode* lchild, *rchild;
}HUNODE;
int findSmallData(HUNODE pArray[] ,int n,int p1, int* p2){
int index = 0;
int fir_small = 0xffff, sec_small = 0xffff;
if(pArray == NULL){
return 1;
}
for(index = 0; index < n; index++){
if(pArray[index] != NULL){
if(pArray[index]->data < fir_small){
sec_small = fir_small;
fir_small = pArray[index]->data;
*p2 = *p1;
*p1 = index;
}
else if(pArray[index]->data < sec_small){
sec_small = pArray[index]->data;
*p2 = index;
}
}
}
return 0;
}
HUNODE* createHuTree(int* a, int n) {
int index = 0;
int fir_small = 0, sec_small = 0;
HUNODE *pArray[100];
HUNODE *pNewNode = NULL;
memset(pArray,0,sizeof(HUNODE)*n);
for(index = 0; index < n; index++){
pNewNode = (HUNODE*)malloc(sizeof(HUNODE));
memset(pNewNode,0,sizeof(HUNODE));
pNewNode->data = a[index];
pNewNode->lchild = NULL;
pNewNode->rchild = NULL;
pArray[index] = pNewNode;
}
for(index = 0; index < n-1; index++){
findSmallData(pArray,n,&fir_small,&sec_small);
pNewNode = (HUNODE*)malloc(sizeof(HUNODE));
memset(pNewNode,0,sizeof(HUNODE));
pNewNode->data = pArray[fir_small]->data + pArray[sec_small]->data;
pNewNode->lchild = pArray[fir_small];
pNewNode->rchild = pArray[sec_small];
pArray[fir_small] = NULL;
pArray[sec_small] = pNewNode;
}
return pNewNode;
}
void preOrderHuffMan(HUNODE* root){
if(root){
printf("%d ",root->data);
preOrderHuffMan(root->lchild);
preOrderHuffMan(root->rchild);
}
}
int main(){
int a[4] = {7,5,2,4};
HUNODE* root = NULL;
root = createHuTree(a,4);
preOrderHuffMan(root);
printf("\n");
return 0;
}