是否同一棵二叉搜索树

4 篇文章 0 订阅

描述

给定一个插入序列就可以唯一确定一棵二叉搜索树。然而,一棵给定的二叉搜索树却可以由多种不同的插入序列得到。例如分别按照序列{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

方法一

思路

将被比较的树建立,然后去判断需要比较的序列是否对应着那棵树。判断的方法,循环查找每一个元素,查找过程中,如果经历了没有被标记的结点,就说明这两棵树不一样,因为如果是同一棵数,同一个元素的位置应该相同,其之前的元素也相同且都已存入。查找完成后,将该结点标记。

代码

#include<iostream>
#include<cstdlib>
using namespace std;

typedef struct TreeNode *Tree;
struct TreeNode											//结点 
{
	int data;
	Tree left,right;
	bool flag;
};

Tree NewNode(int v)										//建立一个新结点 
{
	Tree t=(Tree)malloc(sizeof(TreeNode));				//分配空间 
	t->data=v;
	t->left=t->right=NULL;
	t->flag=0;											//标记设为0 
	return t;
}

Tree Insert(Tree t,int v)								//向树中插入元素 
{
	if(!t)
		t=NewNode(v);									//如果是空树,建立新结点存v 
	else
	{
		if(v>t->data)									//插入到右子树中 
			t->right=Insert(t->right,v);
		else if(v<t->data)								//插入到左子树中 
			t->left=Insert(t->left,v);			
	}
	return t;
}

Tree MakeTree(int n)									//建树 
{
	Tree t=NULL;										//树初始化为空 
	for(int i=1;i<=n;i++)
	{
		int v;
		cin>>v;
		t=Insert(t,v);									//插入 
	}
	return t;
}

bool Check(Tree t,int v)								//检查 
{
	if(t->flag)											//被标记了,就肯定是途径的 
	{
		if(v<t->data)
			return Check(t->left,v);
		else if(v>t->data)
			return Check(t->right,v);
		else 
			return 0;
	}
	else
	{
		if(v==t->data)									//没有被标记,肯定就是最后需要查找的那个 
		{
			t->flag=1;
			return 1;
		}
		else
			return 0;
	}
}

bool Judge(Tree t,int n)
{
	int v;
	for(int i=1;i<=n;i++)
	{
		cin>>v;
		if(!Check(t,v))									//判断每一个 
		{
			for(int j=i+1;j<=n;j++)						//!!!将剩余的数据读完 
				cin>>v;
			return 0;
		}
	}
	return 1;
}

void ResetTree(Tree t)									//清除树的标记 
{
	if(t->left)
		ResetTree(t->left);
	if(t->right)
		ResetTree(t->right);
	t->flag=0;
}

void FreeTree(Tree t)									//释放树的空间 
{
	if(t->left)
		FreeTree(t->left);
	if(t->right)
		FreeTree(t->right);
	free(t);
}

int main()
{
	int n,l;
	while(1)
	{
		cin>>n;
		if(n==0)
			return 0;
		cin>>l;
		Tree t=MakeTree(n);
		for(int i=1;i<=l;i++)
		{
			if(Judge(t,n))
				cout<<"Yes"<<endl;
			else
				cout<<"No"<<endl;
			ResetTree(t);
		}
		FreeTree(t);
	}
}

方法二

思路

将比较和被比较的树都建立,比较两棵树是否相同

代码

#include<iostream>
#include<cstdlib>
using namespace std;

typedef struct TreeNode *Tree;
struct TreeNode											//结点 
{
	int data;
	Tree left,right;
	bool flag;
};

Tree NewNode(int v)										//建立一个新结点 
{
	Tree t=(Tree)malloc(sizeof(TreeNode));				//分配空间 
	t->data=v;
	t->left=t->right=NULL;
	t->flag=0;											//标记设为0 
	return t;
}

Tree Insert(Tree t,int v)								//向树中插入元素 
{
	if(!t)
		t=NewNode(v);									//如果是空树,建立新结点存v 
	else
	{
		if(v>t->data)									//插入到右子树中 
			t->right=Insert(t->right,v);
		else if(v<t->data)								//插入到左子树中 
			t->left=Insert(t->left,v);			
	}
	return t;
}

Tree MakeTree(int n)									//建树 
{
	Tree t=NULL;										//树初始化为空 
	for(int i=1;i<=n;i++)
	{
		int v;
		cin>>v;
		t=Insert(t,v);									//插入 
	}
	return t;
}

bool Judge(Tree t1,Tree t2)
{
	if(t1==NULL&&t2==NULL)
		return 1;
	if(t1==NULL||t2==NULL)
		return 0;
	if(t1->data!=t2->data)
		return 0;
	if(Judge(t1->left,t2->left)&&Judge(t1->right,t2->right))
		return 1;
	return 0;
}

void FreeTree(Tree t)									//释放树的空间 
{
	if(t->left)
		FreeTree(t->left);
	if(t->right)
		FreeTree(t->right);
	free(t);
}

int main()
{
	int n,l;
	while(1)
	{
		cin>>n;
		if(n==0)
			return 0;
		cin>>l;
		Tree t1=MakeTree(n);
		for(int i=1;i<=l;i++)
		{
			Tree t2=MakeTree(n); 
			if(Judge(t1,t2))
				cout<<"Yes"<<endl;
			else
				cout<<"No"<<endl;
			FreeTree(t2);
		}
		FreeTree(t1);
	}
}

方法三

思路

对于两个要比较的序列,先比较首元素大小,再分别比较两个序列大于首元素的和小于首元素的
比如 3124和3412比较 3和3相等, 3124分为{12}3{4} ; 3412分为{12}3{4} 12和12相同,4和4相同,所以两个序列相同
对于3124和3241,3和3相等,3124分为{12}3{4} ; 3214分为{21}3{4} ,按照同样的方法比较12和21,1和2不同,两个序列不同

代码

有点问题,暂时没发现在哪儿。求帮忙看看 orz

#include<iostream>
using namespace std;

int n;

bool judge(int a[],int b[],int l)
{
	int ab[11],as[11],bb[11],bs[11];
	int posab=0,posas=0,posbb=0,posbs=0;
	if(l==0)
		return 1;
	if(a[1]!=b[1])
		return 0;
	if(l==1)
		return 1;
	for(int i=2;i<=l;i++)
	{
		if(a[i]>a[1])
			ab[++posab]=a[i];
		else if(a[i]<a[1])
			as[++posas]=a[i];
	}
	for(int i=2;i<=l;i++)
	{
		if(b[i]>b[1])
			bb[++posbb]=b[i];
		else if(b[i]<b[1])
			bs[++posbs]=b[i];
	}
	if(posab!=posbb||posas!=posbs)
		return 0;
	if(judge(ab,bb,posab)&&judge(as,bs,posas))
		return 1;
	return 0;
}

int main()
{
	while(1)
	{	
		cin>>n;
		if(n==0)
			return 0;
		int a[11],b[11];
		for(int i=1;i<=n;i++)
			cin>>a[i];
		int l;
		cin>>l;
		for(int i=1;i<=l;i++)
		{
			for(int j=1;j<=n;j++)
				cin>>b[j];
			if(judge(a,b,n))
				cout<<"Yes"<<endl;
			else
				cout<<"No"<<endl;
		}
	}
}
### 回答1: 题目描述: 给定两个整数序列,分别表示两棵二叉搜索树的中序遍历序列,判断这两棵二叉搜索树是否一棵。 解题思路: 二叉搜索树的中序遍历序列是有序的,因此我们可以将两个序列排序后比较是否相等。如果相等,则说明两棵二叉搜索树相同。 代码实现: ``` bool isSameBST(vector<int>& seq1, vector<int>& seq2) { if (seq1.size() != seq2.size()) { return false; } sort(seq1.begin(), seq1.end()); sort(seq2.begin(), seq2.end()); return seq1 == seq2; } ``` 时间复杂度:$O(nlogn)$ 空间复杂度:$O(n)$ 其中,$n$ 表示二叉搜索树中节点的个数。 ### 回答2: 题目描述: 给定两个长度相等的整型数组,判断它们是否对应同一棵二叉搜索树的后序遍历序列。 解题思路: 本题的难点在于如何判断两个数组是否对应同一棵二叉搜索树的后序遍历序列。 首先,我们知道二叉搜索树的后序遍历序列的最后一个元素为根节点,而且比根节点小的元素在左子树中,比根节点大的元素在右子树中。 因此,我们可以将数组前面的元素分为左子树和右子树两个部分,然后在子树中递归检查,直到只剩下一个元素为止。 具体实现上,我们可以先找到根节点,然后将数组根据根节点分为左右两个子数组,再递归检查左右子树是否对应同一棵二叉搜索树的后序遍历序列。 如果左右子树都满足条件,那么说明整个数组对应同一棵二叉搜索树的后序遍历序列,否则不是。 Code: ### 回答3: 题目描述: 给定两个长度为n的数组,按照给定的顺序插入到一棵空的二叉搜索树中,判断两个数组是否最终构建了同一棵二叉搜索树。 思路分析: 本题是一道典型的二叉搜索树的应用题,题目要求我们判断两个经过插入操作后最终构建的树是否相等。 首先回顾一下二叉搜索树的特点:对于一个二叉搜索树,它的左子树中的所有节点的值都小于根节点的值,而它右子树中的所有节点的值都大于根节点的值。 对于这个问题来说,我们可以在构建二叉搜索树的同时统计每个节点的左子树大小来判断两棵树是否相同。 具体来说,对于每个节点,我们可以统计它左子树中的节点数目,然后每插入一个节点就更新一遍。当构建完毕后,每个节点的左子树大小就都已经确定了。然后判断两个构建好的树中所有节点的左子树大小是否都一一对应即可。 代码实现: 为了方便起见,我们可以先实现一个二叉搜索树的插入操作,然后再利用这个代码来对两个输入排序数组进行构建。 //二叉搜索树结构体 struct TreeNode { int val; int left_size;//记录左子树大小的变量 TreeNode *left; TreeNode *right; TreeNode(int x) : val(x), left_size(0), left(NULL), right(NULL) {} }; TreeNode* insert(TreeNode* root, int val){ if(!root) return new TreeNode(val); if(val < root->val){ root->left = insert(root->left, val); ++root->left_size;//更新左子树大小 } else root->right = insert(root->right, val); return root; } 然后我们在主函数中对两个输入数组进行插入操作,之后再进行比较即可。 bool isSameTree(vector<int>& nums1, vector<int>& nums2) { if(nums1.size() != nums2.size()) return false; TreeNode* root1 = nullptr;//第一棵树的根节点 TreeNode* root2 = nullptr;//第二棵树的根节点 for(int i=0; i<nums1.size(); ++i){ root1 = insert(root1, nums1[i]); root2 = insert(root2, nums2[i]); } function<bool(TreeNode*, TreeNode*)> dfs = [&](TreeNode* node1, TreeNode* node2){ if(node1 == nullptr && node2 == nullptr) return true; else if(node1 == nullptr || node2 == nullptr) return false; else if(node1->val != node2->val || node1->left_size != node2->left_size) return false; else return dfs(node1->left, node2->left) && dfs(node1->right, node2->right); }; return dfs(root1, root2); } 时间复杂度:插入节点的时间复杂度是O(logn),所以整体时间复杂度是O(nlogn)。 空间复杂度:树的深度最大为O(n),所以空间复杂度也是O(n)。 参考链接: https://leetcode-cn.com/problems/same-tree/description/ https://leetcode-cn.com/problems/two-sum-iv-input-is-a-bst/description/
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值