按层打印二叉树（去哪儿网2017春招真题）

题目描述：

给定一棵二叉树的前序（根、左、右）和中序（左、根、右）的打印结果，输出此二叉树按层（从左往右）打印结果。

例如一棵二叉树前序：1 2 4 5 3；中序：4 2 5 1 3。可以构建出下图所示二叉树：

按层打印的结果则为：1 2 3 4 5。

分析：1，根据先序遍历和中序遍历构建一棵新树。2，按层次遍历这个树

import java.util.*;
public class levelTraversal {
    TreeNode root ;
    TreeNode treeNode ;
//在中序遍历中找根节点
    public int findPos(int[] inOrderT, int val){
        for(int i=0;i<inOrderT.length;i++)
            if(val == inOrderT[i])return i;//在中序遍历中找出子树中的根节点
            return 0;
    }
// 按层次打印
    public void levelTraversal(TreeNode root){
        if (root==null) {
            return;
        }
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        queue.offer(root);
        TreeNode node ;
        while(queue.size()!=0){
            node = queue.poll();
            System.out.print(node.val);
            if(node.left!=null)queue.offer(node.left);
            if(node.right!=null)queue.offer(node.right);
        }
    }
//创建一棵新树
    public TreeNode creatTree(int[] firstT, int fleft, int fright, int[] inOrderT, int ileft, int iright){
        if(fleft>fright || ileft>iright) return null;
        TreeNode treeNode = new TreeNode(firstT[fleft]);//每次新建子树根节点作为父树的孩子节点
        int i = findPos(inOrderT, firstT[fleft]);//找到中序遍历的根节点
        treeNode.left = creatTree(firstT, fleft+1, fleft+i-ileft, inOrderT, ileft, i-1);
        treeNode.right = creatTree(firstT, fleft+i-ileft+1, fright, inOrderT, i+1, iright);
        return treeNode;
    } 

    public static void main (String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int[] a = new int[n];
        int[] b = new int[n];
        for(int i = 0;i<n;i++){
            a[i] = sc.nextInt();
        }
        for(int i = 0;i<n;i++){
            b[i] = sc.nextInt();
        }
        levelTraversal lt = new levelTraversal();
        //TreeNode root=lt.newRoot(a);
        TreeNode root=lt.creatTree(a,0,a.length-1,b,0,b.length-1);
        lt.levelTraversal(root);
    }
}
//一个树节点
class TreeNode{
    int val;
    TreeNode left=null;
    TreeNode right=null;
    public TreeNode(int val){
        this.val = val;
    }
}