L2-011 玩转二叉树 - java

谢谢啊sir

已于 2022-08-12 18:29:26 修改

阅读量182

点赞数

分类专栏： pta 文章标签： java 深度优先算法

于 2022-08-12 18:27:10 首次发布

本文链接：https://blog.csdn.net/weixin_52136008/article/details/126132083

版权

pta 专栏收录该内容

161 篇文章 27 订阅

订阅专栏

L2-011 玩转二叉树

时间限制
400 ms
内存限制
64 MB

题目描述：

给定一棵二叉树的中序遍历和前序遍历，请你先将树做个镜面反转，再输出反转后的层序遍历的序列。所谓镜面反转，是指将所有非叶结点的左右孩子对换。这里假设键值都是互不相等的正整数。

输入格式：
输入第一行给出一个正整数N（≤30），是二叉树中结点的个数。第二行给出其中序遍历序列。第三行给出其前序遍历序列。数字间以空格分隔。

输出格式：
在一行中输出该树反转后的层序遍历的序列。数字间以1个空格分隔，行首尾不得有多余空格。

输入样例：
7
1 2 3 4 5 6 7
4 1 3 2 6 5 7

输出样例：
4 6 1 7 5 3 2

给定二叉树的前序以及中序遍历

求出给定二叉树的镜像反转的层次遍历

镜像反转的层次遍历

也就是根右左

emmmmmmm

可以dfs先递归右子树再去递归左子树

每次递归之前去记录根节点

import java.io.*;
import java.math.*;
import java.util.*;

public class Main
{
	static int N = 30;
//	存储中序 以及 前序
	static int zhong[] = new int[N + 10], qian[] = new int[N + 10];
//	存储镜像层次遍历的结果
	static ArrayList<Integer> tree[] = new ArrayList[N + 10];

//	去在中序中找根节点在哪里 
//		将这段分为两份 
//			一份左节点
//			一份右节点
	static int find(int l, int r, int x)
	{
		for (int i = l; i <= r; i++)
		{
			if (zhong[i] == x)
				return i - l;
		}
		return -1;
	}

	static void dfs(int ql, int qr, int zl, int zr, int ceng)
	{
//		前序 或者 中序 的 左边大于右边
//			这样就不是一颗树
		if (ql > qr || zl > zr)
			return;

//		当前这层添加根节点
		tree[ceng].add(qian[ql]);
//		去找分界点在中序的位置
		int k = find(zl, zr, qian[ql]);

//		先去递归右子树
		dfs(ql + k + 1, qr, zl + k + 1, zr, ceng + 1);
//		再去递归左子树
		dfs(ql + 1, ql + k, zl, zl + k - 1, ceng + 1);
	}

	public static void main(String[] args)
	{
		int n = sc.nextInt();
		for (int i = 1; i <= n; i++)
			tree[i] = new ArrayList<Integer>();

		for (int i = 1; i <= n; i++)
			zhong[i] = sc.nextInt();
		for (int i = 1; i <= n; i++)
			qian[i] = sc.nextInt();

		dfs(1, n, 1, n, 1);

//		存储是否出现过一个
		boolean f = false;
		for (int i = 1; i <= n; i++)
		{
			for (int j : tree[i])
			{
				if (f)
					out.print(" ");
				out.print(j);
				f = true;
			}
		}

		out.flush();
		out.close();
	}

	static Scanner sc = new Scanner(System.in);
	static PrintWriter out = new PrintWriter(System.out);
}