1. 程式人生 > >04-樹4 是否同一棵二叉搜索樹(25 分)

04-樹4 是否同一棵二叉搜索樹(25 分)

tom RM AS null 二叉 type treenode 不同 tree

給定一個插入序列就可以唯一確定一棵二叉搜索樹。然而,一棵給定的二叉搜索樹卻可以由多種不同的插入序列得到。例如分別按照序列{2, 1, 3}和{2, 3, 1}插入初始為空的二叉搜索樹,都得到一樣的結果。於是對於輸入的各種插入序列,你需要判斷它們是否能生成一樣的二叉搜索樹。

輸入格式:

輸入包含若幹組測試數據。每組數據的第1行給出兩個正整數N (10)和L,分別是每個序列插入元素的個數和需要檢查的序列個數。第2行給出N個以空格分隔的正整數,作為初始插入序列。最後L行,每行給出N個插入的元素,屬於L個需要檢查的序列。

簡單起見,我們保證每個插入序列都是1到N的一個排列。當讀到N為0時,標誌輸入結束,這組數據不要處理。

輸出格式:

對每一組需要檢查的序列,如果其生成的二叉搜索樹跟對應的初始序列生成的一樣,輸出“Yes”,否則輸出“No”。

輸入樣例:

4 2
3 1 4 2
3 4 1 2
3 2 4 1
2 1
2 1
1 2
0

輸出樣例:

Yes
No
No

我的答案:
  1 #include <stdio.h>
  2 #include <stdlib.h>
  3 #include <unistd.h>
  4 
  5 typedef struct TreeNode *BinTree;
  6 struct TreeNode {
  7     int Data;
8 BinTree Left; 9 BinTree Right; 10 int Flag; 11 }; 12 13 BinTree Insert(int X, BinTree BST) 14 { 15 if(!BST) { 16 BST = (BinTree)malloc(sizeof(struct TreeNode)); 17 BST->Data = X; 18 BST->Left = BST->Right = NULL; 19 BST->Flag = 0
; 20 } else { 21 if(X < BST->Data) 22 BST->Left = Insert(X, BST->Left); 23 else if(X > BST->Data) 24 BST->Right = Insert(X, BST->Right); 25 } 26 return BST; 27 } 28 29 BinTree MakeTree(int num) 30 { 31 BinTree S = NULL; 32 int i, data; 33 for(i=0;i<num;i++) { 34 scanf("%d", &data); 35 S = Insert(data, S); 36 } 37 return S; 38 } 39 40 void ResetT(BinTree S) 41 { 42 if(S->Left) ResetT(S->Left); 43 if(S->Right) ResetT(S->Right); 44 S->Flag = 0; 45 } 46 47 void FreeTree(BinTree S) 48 { 49 if(S->Left) FreeTree(S->Left); 50 if(S->Right) FreeTree(S->Right); 51 free(S); 52 } 53 54 void InOrderTraversal(BinTree BT) 55 { 56 if(BT) { 57 InOrderTraversal(BT->Left); 58 printf("%d ", BT->Data); 59 InOrderTraversal(BT->Right); 60 } 61 } 62 63 int Check(BinTree T, int data) 64 { 65 if(T->Flag) { 66 if(data < T->Data) 67 return Check(T->Left, data); 68 else if(data > T->Data) 69 return Check(T->Right, data); 70 else 71 return 0; /* 錯誤? */ 72 } else { 73 if(data == T->Data) { 74 T->Flag = 1; 75 return 1; 76 } else 77 return 0; 78 } 79 } 80 81 int Judge(BinTree T, int N) 82 { 83 int i, data, flag = 0; 84 scanf("%d", &data); 85 if(data != T->Data) 86 flag = 1; 87 else 88 T->Flag = 1; 89 90 for(i=1;i<N;i++) { 91 scanf("%d", &data); 92 if((!flag)&&(!Check(T, data))) 93 flag = 1; 94 } 95 if(flag) 96 return 0; 97 else 98 return 1; 99 } 100 101 int main() 102 { 103 int N, L; 104 int i, j, data; 105 BinTree SourceTree, CompareTree; 106 while(1) { 107 scanf("%d", &N); 108 if(!N) break; 109 scanf("%d", &L); 110 SourceTree = MakeTree(N); 111 // printf("InOrder:"); 112 // InOrderTraversal(SourceTree); 113 // printf("\n"); 114 for(i=0;i<L;i++) { 115 if(Judge(SourceTree, N)) 116 printf("Yes\n"); 117 else 118 printf("No\n"); 119 ResetT(SourceTree); 120 } 121 FreeTree(SourceTree); 122 } 123 return 0; 124 }

04-樹4 是否同一棵二叉搜索樹(25 分)