是否同一棵二叉搜索树 (25分)

给定一个插入序列就可以唯一确定一棵二叉搜索树。然而,一棵给定的二叉搜索树却可以由多种不同的插入序列得到。例如分别按照序列{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<algorithm>
#include<cstring>
using namespace std;
int a[1500],b[1500];
void dfs(int i,int t){
	if(a[i]==-1){
		a[i]=t;
		return ;
	}
	if(t<a[i]) dfs(2*i,t);
	else dfs(2*i+1,t);
}
void add(int i,int t){
	if(b[i]==-1){
		b[i]=t;
		return ;
	}
	if(t<b[i]) add(2*i,t);
	else add(2*i+1,t);
}
int main ()
{
	int n,l,t;
	while(1){	
	memset(a,-1,sizeof(a));
	memset(b,-1,sizeof(b));
	 scanf("%d",&n);
	 if(n==0)
	 return 0;
	 scanf("%d",&l);
		for(int i=0;i<n;i++){
		scanf("%d",&t);
		dfs(1,t);
	}
	for(int i=0;i<l;i++){
		memset(b,-1,sizeof(b));
		int ans=0;
		int sum=0;
		for(int j=0;j<n;j++){
			scanf("%d",&t);
			add(1,t);
		}
		for(int j=0;;j++){
			if(b[j]>0){
				ans++;
				if(a[j]==b[j])
				sum++;
			}
			if(ans==n)
			break;
			
		}
		if(sum==n)
		printf("Yes\n");
		else
		printf("No\n");
	  }
	}
	return 0;
}
  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
### 回答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、付费专栏及课程。

余额充值