04-樹4 是否同一棵二叉搜尋樹
阿新 • • 發佈:2019-02-04
思路:看的何老師的視訊敲的。
先建立正確的樹,然後將查詢的樹在正確的樹上查詢是否相等
怎麼查詢?在原樹上的所有結點上設定個標記,查詢時,每當進去一個結點時,看它是否按正確的軌跡進行走的,正確的話就是走過的結點都是標記過的,如果當前沒有被標記,說明就是當前位置,如果不相等,說明就不是原樹了!
程式碼:
#include <cstdio> #include <cstdlib> #include <algorithm> typedef struct TreeNode *Tree; struct TreeNode{ int v; Tree Left,Right;//左右子結點 int flag; }; void ResetT(Tree T){//重置結點的標記0 if(T->Left) ResetT(T->Left); if(T->Right) ResetT(T->Right); T->flag = 0; } void FreeTree(Tree T){//釋放空間 if(T->Left) FreeTree(T->Left); if(T->Right) FreeTree(T->Right); free(T); } Tree NewNode(int v){//建立首結點 Tree T = (Tree)malloc(sizeof(struct TreeNode));//注意sizeof中不能放指標 T->v = v; T->Left = T->Right = NULL; T->flag = 0; return T; } Tree Insert(Tree T,int v){//插入結點 if(!T) T = NewNode(v);//空樹建立新結點 else{//插入 if(v > T->v) T->Right = Insert(T->Right,v); else T->Left = Insert(T->Left,v); } return T; } Tree MakeTree(int N){//建樹 int v; Tree T = NULL; for(int i=0;i<N;i++){ scanf("%d",&v); T = Insert(T,v); } return T; } int check(Tree T,int v){//尋找當前結點是否在正確位置 if(T->flag){ if(v > T->v) return check(T->Right,v); if(v < T->v) return check(T->Left,v); else return 0;//重現重複結點 }else{ if(v == T->v){//位置正確 T->flag = 1;//標記位置 return 1;// }else return 0;//位置不正確 } } int judge(Tree T,int N){//判斷是否同一棵樹 int v,flag = 1; for(int i=0;i<N;i++){ scanf("%d",&v); if(flag && !check(T,v)) flag = 0; } if(flag) return 1; return 0; } int main(){ int n,m; Tree T; while(scanf("%d",&n) && n){ scanf("%d",&m); T = MakeTree(n);//建樹 ResetT(T);//標記 for(int i=0;i<m;i++){ if(judge(T,n)) printf("Yes\n"); else printf("No\n"); ResetT(T);//標記 } FreeTree(T);//釋放空間 } return 0; }