重建二叉樹


知識:在先序遍歷中,第一個結點就是二叉樹的根節點;而在中序遍歷中,根節點必然將中序序列分割成兩個子序列,前一個子序列就是根節點的左子樹的中序序列,后一個是根節點的右子樹的中序序列。同樣,給定后序序列和中序序列,按層次序列和中序序列可以也可以唯一確定一棵二叉樹。但是,如果知道二叉樹的先序序列和后序序列,則無法唯一確定一棵二叉樹。

例子:如給定先序序列和中序序列,建立一棵二叉樹,給出重建二叉樹的算法:

算法思想:使用遞歸思想。從先序序列中找出第一個結點,即為根節點。掃描中序序列,找到根節點的值,然后可以得到左右子樹的中序序列點的集合。同時通過左右子樹的大小,對先序序列進行偏移,亦可以得到左右子樹的先序遍歷序列。遞歸對左右子樹進行建樹。
算法步驟:

若二叉樹不為空

1、傳入兩對指針,分別代表先序和中序序列的范圍;

2、從先序列中取根節點的值;

3、創建新樹的根節點,給根節點賦值,並將左右指針域初始化為NULL;

4、考慮邊界條件:當序列只有一個節點的時候,如果不一樣,則非法輸入,否則返回新樹的根指針

5、遍歷中序序列,找出與先序序列中的根節點相同的節點,即根節點;

6、計算左右子樹各自起始和結尾指針;

7、遞歸創建左右子樹。

代碼如下:

 1 typedef struct BinTreeNode
 2 {
 3     int data;
 4     BinTreeNode *lc;
 5     BinTreeNode *rc;
 6 }BinTreeNode,*BinTree;
 7 
 8 BinTreeNode * Construct(int *preorder,int *inorder,int length)
 9 {
10     if(preorder==NULL||inorder==NULL||length<=0)
11         return NULL;
12     return Constructcore(preorder,preorder+length-1,inorder,inorder+length-1);
13 }
14 
15 BinTreeNode *Constructcore(int *preStart,int *preEnd,int *inStart,int *inEnd)
16 {
17     int rootVal = preStart[0];//獲取根結點的值
18     BinTreeNode root = new BinTreeNode();//創建結點
19     root->data = rootVal;
20     root-lc=root->rc=NULL;
21     if(preStart == preEnd)//邊界條件,只有一個結點
22     {
23         if(inStart == inEnd && *preStart == *inStart)
24             return root;
25         else
26             throw std::exception("Invalid input.");//非法輸入
27     }
28 
29     //在中序遍歷序列中尋找根節點
30     int *inRoot = inStart;
31     while(inRoot<=inEnd && *inRoot != rootVal)//rootVal是前序遍歷的根節點
32         ++inRoot;//直到找到根節點或者到中序序列結尾
33     if(inRoot == inEnd && *inRoot!=rootVal)
34         throw std::exception("Invalid input.");//當到達中序序列的結尾還沒找到根節點,則說明輸入的序列非法
35     int leftlen = inRoot-inStart;//指針的算術元素(減法)得到偏移量
36     int *leftPreEnd = preStart+len;
37 
38     //構建子樹
39     if(leftlen>0)
40     {
41         root-lc = Constructcore(preStart+1,leftPreEnd,inStart.inRoot-1);
42     }
43     if(leftlen<preEnd - preStart)
44     {
45         root-rc = Constructcore(leftPreEnd+1,preEnd,inRoot+1,inEnd);
46     }
47 
48     return root;
49 }

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM