非遞歸建立二叉樹
前言
使用遞歸(Recursion)建立二叉樹(Binary Tree)的非順序存儲結構(即二叉鏈表),可以簡化算法編寫的復雜程度,但是遞歸效率低,而且容易導致堆棧溢出,因而很有必要使用非遞歸算法。
引入
無論是單鏈表還是二叉樹,創建時要解決問題就是關系的建立,即單鏈表中前驅節點與當前節點的關系和二叉樹中父節點與子節點的關系。
首先,思考一下建立單鏈表的過程,為了使鏈表各個節點連接起來,在創建當前節點(q)的時候,需借助一個指針(p)指向前一個節點,然後p->next = q。
由此推廣至二叉樹,把二叉樹每一層比作是鏈表的節點,接著借助一個指針列表(parent_list)存放父層的所有節點,
分析
引入中提出了創建二叉樹的整體思想,同時也拋出一個問題,如何建立父節點與子節點的關系?
下圖為一棵普通的二叉樹,下面對其進行一些處理。
首先,將其補全,用#代表空節點,補全規則為:將只有一個或沒有子節點的節點(空節點除外),用空節點補全為兩個子節點。
為便於後續分析,將其節點左結構化並去掉關系線。
現在,回顧一下引入中提到的“把二叉樹每一層比作鏈表的節點”,而建立單鏈表每次都只涉及兩個節點,因而下面每次分析都只涉及兩層。
其中,規定第二層為當前層,第一層為當前層的父層,且當前層為下次分析的父層。
在規定,父層為一個數組p[i](i為父層節點數,後同),當前層為數組q[j]。
下圖為第一次,選中層的節點標記為深灰底色。
可以容易看出,當前層的兩個節點與父層節點的關系:
p[0]->next = q[0]
p[0]->next = q[1]
為其添加關系線,然後再看下一次。
同樣,其關系如下:
p[0]->next = q[0]
p[0]->next = q[1] = #
p[1]->next = q[2]
p[2]->next = q[3]
由前兩次不難得出,j/2 = i (註意這裏 / 運算結果只取整數),這個結果和完全二叉樹的性質相同,但是註意這裏不是一棵完全二叉樹。
接著看下一次。
如圖所示,為了建立正確的二叉樹關系,父層的節點一定不能為空節點。
總觀整個結構,可以得出一個規律(該規律可用於動態改變父層列表,以減少內存占用):父層節點數(不包括空節點)的兩倍恰好為當前層的節點數(包括空節點)。
此時,還有個小問題,便是判斷當前節點是左子樹(Left Subtree)還是右子樹(Right Subtree)?
這裏解決方法很簡單,便是計算 j % 2 ,若為0則為左子樹,否則為右子樹。
實現
現在,理清一下思路:
1.以從上到下、從左到右的順序創建二叉樹,因此有兩層循環。
2.有個父層列表(parent_list)用於存放父層所有節點的地址,且外層循環一次就更新一次(即讓父層列表等於當前層列表,代碼中為tmp_list),同時釋放舊父層列表。
3.內層循環創建每一層的節點,若輸入數據不為“#”則不為空節點,然後申請內存空間並賦值,再根據上述論述進行父節點和當前節點(current_node)建立關系。
下面給出代碼:
1 #include <stdio.h> 2 #include <malloc.h> 3 4 // 布爾類型 5 typedef enum {FALSE=0,TRUE=1} bool; 6 // 用於標識當前建立左子樹還是右子樹 7 typedef enum {LEFT=0,RIGHT=1} flag; 8 // 節點存放數據的類型 9 typedef char data_type; 10 // 二叉樹節點類型 11 typedef struct node { 12 data_type data; 13 struct node *left_subtree, 14 *right_subtree; 15 } node , *bin_tree; 16 17 bool create_bin_tree(bin_tree *root) 18 { 19 /* 創建根節點 */ 20 data_type data = ‘\0‘; 21 scanf("%c",&data); 22 if(data == ‘#‘){ 23 return FALSE; // 根節點為空,創建失敗 24 } 25 else 26 { 27 *root = (node*)malloc(sizeof(node)); 28 (*root)->data = data; 29 (*root)->left_subtree = NULL; 30 (*root)->right_subtree = NULL; 31 } 32 33 /* 創建非根節點 */ 34 // 存放父層的節點列表 35 node **parent_list = (node**)malloc(sizeof(node*)); 36 parent_list[0] = *root; 37 // 父節點個數 38 int parent_amount = 1; 39 40 while(1) 41 { 42 // 當前節點個數,設置為父節點個數的兩倍 43 int current_amount = parent_amount * 2; 44 // 創建臨時列表存放當前深度的節點 45 node **tmp_list = (node**)malloc(sizeof(node*) 46 * current_amount); 47 // 用於記錄當前深度節點非空節點個數 48 int count = 0; 49 // 創建當前層次的所有節點 50 int j = 0; 51 for(;j < current_amount;++j) 52 { 53 data = ‘\0‘; 54 scanf("%c",&data); 55 if(data != ‘#‘) // 不為空節點 56 { 57 // 新建節點並賦值 58 node *current_node = (node*)malloc(sizeof(node)); 59 current_node->data = data; 60 current_node->left_subtree = NULL; 61 current_node->right_subtree = NULL; 62 // 加入到臨時列表中 63 tmp_list[count] = current_node; 64 // 非空節點數加1 65 count++; 66 // 與父節點建立關系 67 if(j%2 == LEFT) 68 { 69 (parent_list[j/2])->left_subtree = current_node; 70 } 71 else 72 { 73 (parent_list[j/2])->right_subtree = current_node; 74 } 75 } 76 } // for循環結束 77 78 // 釋放父層列表 79 free(parent_list); 80 // 更新父層列表 81 parent_list = tmp_list; 82 // 若非空節點數為0,則停止創建 83 if(count == 0) break; 84 } 85 return TRUE; 86 }
上述代碼中,由於根節點較特殊且需要傳出地址,為了降低代碼編寫復雜程度,因而獨立於內層循環。
後話
寫文章除了用於記錄外,其實也無形中能理清想問題的思路。如寫這篇文章前,代碼雖實現了,但總感覺思路很亂。而文章寫完了,便也豁然開朗了。
非遞歸建立二叉樹