根据后序中序序列构建二叉树,按层次打印


import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;

public class MidAfterTreeCalLevel {

	/*
	 * 7
	*2 3 1 5 7 6 4(后序)
	*1 2 3 4 5 6 7(中序)
	*打印层次遍历:4 1 6 3 5 7 2
	*基本思路:后序序列的最后一个元素为根,再中序序列中找到根的索引,
	*左边即为左子树,右边极为右子树,分别对左右子树递归构建二叉树
	 */
	public Node calTree(int[] mid, int[] after) {

		if (mid.length == 0) {//结点为null
			return null;
		}
		if (mid.length == 1) {//当前结点为叶子结点
			Node node = new Node(mid[0]);
			return node;
		}

		int rootData = after[after.length - 1];
		Node root = new Node(rootData);//根节点

		int middle = 0;
		for (int i = 0; i < mid.length; i++) {
			if (rootData == mid[i]) {
				middle = i;//根节点在中序的索引
			}
		}

		int[] leftMid = new int[middle];
		int[] leftAfter = new int[middle];
		for (int i = 0; i < middle; i++) {
			leftMid[i] = mid[i];
			leftAfter[i] = after[i];
		}

		int[] rightMid = new int[mid.length - middle - 1];
		int[] rightAfter = new int[mid.length - middle - 1];
		for (int i = 0; i < mid.length - middle - 1; i++) {
			rightMid[i] = mid[i + middle + 1];
			rightAfter[i] = after[i + middle];
		}
        //中序左子树,后序左子树
		root.left = calTree(leftMid, leftAfter);
		//中序右子树,后序右子树
		root.right = calTree(rightMid, rightAfter);

		return root;
	}

	public static void main(String[] args) {

		MidAfterTreeCalLevel mtc = new MidAfterTreeCalLevel();
		Scanner in = new Scanner(System.in);
		int n = in.nextInt();
		int[] after = new int[n];
		int[] mid = new int[n];
		for (int i = 0; i < n; i++) {
			after[i] = in.nextInt();
		}
		for (int i = 0; i < n; i++) {
			mid[i] = in.nextInt();
		}

		Node root = mtc.calTree(mid, after);
		mtc.print(root);
	}
	
	private void print(Node root) {//使用队列层次打印二叉树
		Queue<Node> queue = new LinkedList<Node>();
		StringBuilder sb = new StringBuilder();
		queue.add(root);
		while (!queue.isEmpty()) {
			Node node = queue.poll();
			sb.append(node.data + " ");
			if (node.left != null) {
				queue.add(node.left);
			}
			if (node.right != null) {
				queue.add(node.right);
			}
		}
		System.out.println(sb.toString().trim());
	}
	
	private class Node {//自定义结点
		int data;
		Node left = null;
		Node right = null;
		public Node(int data) {
			this.data = data;
		}
	}
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

算法小生Đ

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值