二叉搜索树【浙江大学】★

查看原题目请点我这里

解题思路

首先进行根据输入的数据建立二叉排序树，然后将后面的输入依次与最开始的树对比是否是相同的树。关于是否是两棵相同的树，LeetCode上面有道这样的题，好多大神写的代码简洁又高效，我刚好想到这上面的这道题，直接搬过来用了。

版本2

#include <iostream>
#include <vector>
#include <string>
#include <cmath>
#include <algorithm>
#include <queue>
#include <cstdio>
#include <cctype>
#include <unordered_map>
#include <map>
using namespace std;
const int N = 1005;
typedef pair<int, string> PII;
struct tree{
	char val;
	tree* left, *right;
	tree(char x){
		val = x;
		left = right = nullptr;
	}
};
void create(tree* &root, char val){
	if(!root){
		root = new tree(val);
	}else if(val > root->val)
		create(root->right, val);
	else 
		create(root->left, val);
} 
bool issame(tree* p, tree* q){
	if(!p || !q) return !p && !q;
	if(p->val != q->val) return false;
	return issame(p->left, q->left) && issame(p->right, q->right);
}
int main(){
	int n, x;
	while(cin>>n){
		if(!n) break;
		tree* root = nullptr;
		string str;
		cin>>str;
		for(auto c: str){ 
			create(root, c);
		}
		for(int i = 0; i < n; i++){
			cin>>str;
			tree* rt = nullptr;
			for(auto c: str){ 
				create(rt, c);
			}
			if(issame(root, rt)) cout<<"YES"<<endl;
			else cout<<"NO"<<endl;
		}	
	} 
	return 0;
}

版本1

#include<cstdio>
#include<cstring>
struct node{
	int data;
	node *lchild,*rchild;
}; 
char str[15],s[15];
void insert(node* &root,int data){
	if(root==NULL){
		root=new node;
		root->data=data;
		root->lchild=root->rchild=NULL;
		return;
	}
	if(data > root->data) 
		insert(root->rchild,data);
	else 
		insert(root->lchild,data);
}
bool sameTree(node* p,node* q){
	if(p==0) return q==0;
	if(q==0) return false;
	return p->data==q->data&&sameTree(p->lchild,q->lchild)&&sameTree(p->rchild,q->rchild);
}
int main(){
	int n;
	while(scanf("%d",&n)!=EOF&&n!=0){
		scanf("%s",str);
		int len=strlen(str);
		node *root=NULL;
		for(int i=0;i<len;i++){
			insert(root,str[i]-'0');
		}
		for(int i=0;i<n;i++){
			scanf("%s",s);
			node *root1=NULL;
			int len1=strlen(s);
			for(int i=0;i<len1;i++){
				insert(root1,s[i]-'0');
			}
			if(sameTree(root,root1))
				printf("YES\n");
			else 
				printf("NO\n");
		}	
	}
	return 0;
}