Fixed huffman edge case failure in C. (#661)

Gathros · leios · commit 4bf9a8ab5be9 · 2019-12-24T07:35:49.000+09:00
diff --git a/contents/huffman_encoding/code/c/huffman.c b/contents/huffman_encoding/code/c/huffman.c
@@ -134,6 +134,15 @@ struct tree* generate_tree(const char* str){
 }
 }
 
+ if (heap.length == 1){
+ struct tree* leaf = heap_pop(&heap);
+ struct tree* root = calloc(0, sizeof(struct tree));
+ root->left = leaf;
+ root->count = leaf->count;
+ heap_free(&heap);
+ return root;
+ }
+
 while (heap.length > 1){
 struct tree* left = heap_pop(&heap);
 struct tree* right = heap_pop(&heap);

-Original file line number
+Diff line change
+ }
+ }
 +if (heap.length==1){
 +structtree*leaf=heap_pop(&heap);
 +structtree*root=calloc(0, sizeof(structtree));
 +root->left=leaf;
 +root->count=leaf->count;
 +heap_free(&heap);
 +returnroot;
 + }
++
 while (heap.length>1){
 structtree*left=heap_pop(&heap);
 structtree*right=heap_pop(&heap);