根据先序序列和中序序列构造二叉树进行层次遍历

基于遍历先序序列的每个元素分割中序序列为左右子树进行递归操作。

import java.util.HashMap;
import java.util.LinkedList;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Queue;

public class Tree {

	static int[] pretemp;
	private static int maxLevel = 0;

	public class BinarayTreeNode {
		public int m_Value;
		public BinarayTreeNode m_LeftTreeNode = null;
		public BinarayTreeNode m_RigtTreeNode = null;
	}

	public BinarayTreeNode Construct(int[] preorder, int[] inorder) {
		if (preorder == null || inorder == null || inorder.length == 0 || preorder.length == 0) {
			return null;
		}
		int ptr = 0;
		int rootValue = preorder[0];
		BinarayTreeNode rootNode = new BinarayTreeNode();
		rootNode.m_Value = rootValue;

		while (ptr <= inorder.length && inorder[ptr] != rootValue) {
			ptr++;
		}
		pretemp = new int[preorder.length - 1];

		for (int i = 0; i < preorder.length; i++) {
			if (i == 0) {
				continue;
			}
			pretemp[i - 1] = preorder[i];
		}
		int[] inStart = new int[ptr];
		int[] inEnd = new int[inorder.length - ptr - 1];
		int ps = 0;
		for (int j = 0; j < inorder.length; j++) {
			if (j < ptr) {
				inStart[j] = inorder[j];
			}
			if (j > ptr) {
				inEnd[ps] = inorder[j];
				ps++;
			}
		}

		rootNode.m_LeftTreeNode = Construct(pretemp, inStart);
		rootNode.m_RigtTreeNode = Construct(pretemp, inEnd);

		return rootNode;
	}

	// print binarayTree as level mode
	public static void PrintLevelTree(BinarayTreeNode pRoot) {
		Queue<BinarayTreeNode> nodeQueue = new LinkedList<BinarayTreeNode>();
		nodeQueue.add(pRoot);
		while (nodeQueue != null) {
			BinarayTreeNode node = nodeQueue.poll();
			if (node != null)
				System.out.print(node.m_Value);
			if (node != null && node.m_LeftTreeNode != null)
				nodeQueue.add(node.m_LeftTreeNode);
			if (node != null && node.m_RigtTreeNode != null)
				nodeQueue.add(node.m_RigtTreeNode);
			if (node == null) {
				System.out.println();
				break;
			}
		}
	}

	// get current binarayTree level
	private static int getTreeLeverl(BinarayTreeNode pRoot, int level) {
		if (pRoot != null) {
			level++;
		}
		if( maxLevel < level ) maxLevel = level;
		if (pRoot.m_LeftTreeNode != null) {
			getTreeLeverl(pRoot.m_LeftTreeNode, level);
		} 
		if (pRoot.m_RigtTreeNode != null) {
			getTreeLeverl(pRoot.m_RigtTreeNode, level);
		}
		return maxLevel ;
	}

	private static void printTreeLeverl(Map<Integer, String> printInfo, BinarayTreeNode pRoot, int level) {

		String value = printInfo.get(level);
		value = value == null ? "" : value;
		value = value + pRoot.m_Value +" " ;
		printInfo.put(level, value);
		level++;

		if (pRoot.m_LeftTreeNode != null) {
			printTreeLeverl(printInfo, pRoot.m_LeftTreeNode, level);
		}

		if (pRoot.m_RigtTreeNode != null) {
			printTreeLeverl(printInfo, pRoot.m_RigtTreeNode, level);
		}
	}


	public static void main(String[] args) {
		int preorder[] = { 1, 2, 4,9, 7, 0,8,3, 5, 6, 8 }; // 先序
		int inorder[] = { 9, 4, 7,8, 0,2, 1, 5, 3, 8, 6 }; // 中序
		//根据先序中序构造二叉树
		BinarayTreeNode root = new Tree().Construct(preorder, inorder);
		System.out.println("==========");
		int totollevel = getTreeLeverl(root, 0);
		System.out.println("二叉树的深度:"+totollevel);
		System.out.println("==========");
		System.out.print("二叉树的层次遍历:");
		PrintLevelTree(root);
		System.out.println("===========");
		System.out.println("二叉树的层次结构:");
		System.out.println("===========");
		Map<Integer, String> printInfo = new HashMap<Integer, String>();
		printTreeLeverl(printInfo, root, 1);
		for (Entry<Integer, String> map : printInfo.entrySet()) {
			System.out.println(map.getValue());
		}
	}
}

 

  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

喜欢编程的网管

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

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

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

打赏作者

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

抵扣说明:

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

余额充值