根据前序遍历判断二叉搜索树

题意:给你一个二叉树的前序遍历序列,让你判断是不是二叉搜索树(左子树小于根,右子树大于根)或其镜像。如果是,输出后序遍历序列。

开始走了弯路,一直在思考如何建树,结果试了很多种方法都不对,总是在一个新节点该放在哪个节点下出问题。

其实这道题可以和给出前序中序写后序这种题类比,完全可以不建树。根据前序遍历的性质每次可以把根节点分离出来;根据二叉搜索树的性质每次可以把左右子树分开;这样在判断的时候只判断左边的是不是都小于根,右边的是不是都大于根就行了。至于后续序列,在递归左右之后,用b数组保存。

#include <stdio.h>
#include <algorithm>
using namespace std;
int a[1003], n, k = 0, b[1003]; 
int check1(int l, int r)
{
	if(l > r)
	    return 1;
	int root = a[l];//把根节点拿出来 
	int i, j;
	for(i = l + 1 ; i <= r && a[i] < root; i++);//根据二叉搜索树的性质,把左右子树分开 
	for(j = i ; j <= r ; j++)//检查右子树是不是都大于根 
	{
		if(a[j] < root)
		    return 0;
	}
	if(check1(l + 1, i - 1) == 0)//递归地检查左右子树 
	    return 0;
	if(check1(i, r) == 0)
	    return 0;
	b[k++] = a[l];//b保存后序遍历序列 
	return 1;
}
int check2(int l, int r)
{
	if(l > r)
	    return 1;
	int root = a[l];
	int i, j;
	for(i = l + 1 ; i <= r && a[i] >= root; i++);
	for(j = i ; j <= r ; j++)
	{
		if(a[j] >= root)
		    return 0;
	}
	if(check2(l + 1, i - 1) == 0)
	    return 0;
	if(check2(i, r) == 0)
	    return 0;
	b[k++] = a[l];
	return 1;
}
int main()
{
	int i;
	scanf("%d", &n);
	for(i = 1 ; i <= n ; i++)
	    scanf("%d", &a[i]);
	if(n == 1)
	{
		printf("YES\n%d\n", a[1]);
		return 0;
	}
	if(a[1] > a[2])
	{
		if(check1(1, n))
		{
			printf("YES\n");
			for(i = 0 ; i < n - 1 ; i++)
			    printf("%d ", b[i]);
			printf("%d\n", b[n - 1]);
		}
		else
		    printf("NO\n");
	}
	else
	{
		if(check2(1, n))
		{
			printf("YES\n");
			for(i = 0 ; i < n - 1 ; i++)
			    printf("%d ", b[i]);
			printf("%d\n", b[n - 1]);
		}
		else
		    printf("NO\n");
	}
	return 0;
}


  • 5
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
抱歉,我是一名语言模型AI,无法为您提供实际编程操作,但可以为您提供伪代码: 1. 定义二叉搜索树节点类 class TreeNode { int val; TreeNode left; TreeNode right; TreeNode(int val) { this.val = val; this.left = null; this.right = null; } } 2. 实现二叉搜索树的构建方法 public TreeNode buildTree(int[] preorder, int[] inorder, int preStart, int preEnd, int inStart, int inEnd) { if (preStart > preEnd) { return null; } TreeNode root = new TreeNode(preorder[preStart]); int index = inStart; while (index <= inEnd && inorder[index] != root.val) { index++; } int leftSize = index - inStart; root.left = buildTree(preorder, inorder, preStart + 1, preStart + leftSize, inStart, index - 1); root.right = buildTree(preorder, inorder, preStart + leftSize + 1, preEnd, index + 1, inEnd); return root; } 3. 实现二叉搜索树的遍历方法 // 前序遍历 public void preorder(TreeNode root, List<Integer> list) { if (root == null) { return; } list.add(root.val); preorder(root.left, list); preorder(root.right, list); } // 中序遍历 public void inorder(TreeNode root, List<Integer> list) { if (root == null) { return; } inorder(root.left, list); list.add(root.val); inorder(root.right, list); } // 后序遍历 public void postorder(TreeNode root, List<Integer> list) { if (root == null) { return; } postorder(root.left, list); postorder(root.right, list); list.add(root.val); } 4. 将遍历结果写入csv文件 public void writeCsvFile(List<Integer> list, String filename) { try { FileWriter writer = new FileWriter(filename); for (int val : list) { writer.append(Integer.toString(val)); writer.append(','); } writer.flush(); writer.close(); } catch (IOException e) { e.printStackTrace(); } } 5. 主函数中调用上述方法 public static void main(String[] args) { int[] preorder = {3, 9, 20, 15, 7}; int[] inorder = {9, 3, 15, 20, 7}; int[] postorder = {9, 15, 7, 20, 3}; Solution solution = new Solution(); TreeNode root = solution.buildTree(preorder, inorder, 0, preorder.length - 1, 0, inorder.length - 1); List<Integer> preorderList = new ArrayList<>(); solution.preorder(root, preorderList); solution.writeCsvFile(preorderList, "preorder.csv"); List<Integer> inorderList = new ArrayList<>(); solution.inorder(root, inorderList); solution.writeCsvFile(inorderList, "inorder.csv"); List<Integer> postorderList = new ArrayList<>(); solution.postorder(root, postorderList); solution.writeCsvFile(postorderList, "postorder.csv"); }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值