1. 程式人生 > >04-樹4 是否同一棵二叉搜尋樹

04-樹4 是否同一棵二叉搜尋樹

思路:看的何老師的視訊敲的。

先建立正確的樹,然後將查詢的樹在正確的樹上查詢是否相等

怎麼查詢?在原樹上的所有結點上設定個標記,查詢時,每當進去一個結點時,看它是否按正確的軌跡進行走的,正確的話就是走過的結點都是標記過的,如果當前沒有被標記,說明就是當前位置,如果不相等,說明就不是原樹了!

程式碼:

#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;
}