华为OD机试C卷“二叉树给后序和中序遍历输出层序遍历”Java解答

描述

示例

算法思路1

可能有点小问题，就是找左孩子的时候，应该是找中序数组根左面到父根中的字母在后序数组中最靠后的一个，下面代码并未实现这个逻辑，而是有点投机了，定义了 一个size，每当队头元素出队时size就自增，当size与二叉树元素相等时就不在遍历了。所有下述代码可以解决本题，但是不能准确找出左右孩子，欢迎评论区各路大神帮忙修正，万分感谢！！！（毕竟我真的有点菜菜，不知道该怎么记录父根的位置，可能需要树的结构TreeNode，学一下回来下篇一定更新！！！） 

答案1

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        String postorder = scanner.nextLine();
        String inorder = scanner.nextLine();
        char[] post = postorder.toCharArray();
        char [] in = inorder.toCharArray();
        char root = post[post.length-1];
        //System.out.println(root);//整个树的根
        int size=0;
        Queue<Character> queue = new LinkedList<>();//声明一个队列用于层序遍历
        queue.offer(root);//将树根加入队列
        while(!queue.isEmpty()&&size!=post.length){//实现层序遍历的队列逻辑，输出队头时，队头左右孩子入队
            char headChar = queue.remove();
            size++;
            System.out.print(headChar); //队头输出
            char lChild = FindLeftChild(headChar,post,in);
            //System.out.print("左孩子是"+lChild);
            if(lChild!=' '){
                queue.offer(lChild);//左孩子入队
            }
            char rChild = FindRightChild(headChar,post,in);
           // System.out.print("右孩子是"+rChild);
            if(rChild!=' '){
                queue.offer(rChild);//右孩子入队
            }
        }
    }

    private static char FindRightChild(char root, char [] post,char [] in) {//找右孩子
        char rchild=' ';
        for(int i=post.length-1;i>=0;i--){
            if(post[i]==root){//找到当前根
                if(i-1>=0){//根前有字母M
                    int k=getIndex(root,in);//找根在中序数组中的位置
                    for(int j=k;j<in.length;j++){
                        if(in[j]==post[i-1]){//在中序数组中根右边有M
                            rchild = post[i-1];//根的右孩子
                         //   System.out.println("找到右孩子");
                            return rchild;
                        }
                    }
                }
            }
        }
        return rchild;
    }

    private static char FindLeftChild( char root, char[] post, char[] in) {//找根的左孩子
        char lchild=' ';
        //int father = index;//当前元素根的父根
        int k=getIndex(root,in);//找根在中序数组中的位置
       // System.out.println("根在中序数组中是第"+k+"位");
        int m=getIndex(root,post);//找根在后序数组中的位置
       // System.out.println("根在后序数组中是第"+m+"位");
        for(int i=m-1;i>=0;i--){//后序数组从根找
            for(int j=k;j>=0;j--){//中序数组从根找
                if(i>=0){//根在后序数组中前面有元素
                    if(post[i]==in[j]){//后序数组中，根前面的元素在中序数组中根的左面，则为左孩子
                        lchild=post[i];
                       // System.out.println("找到左孩子");
                        return lchild;
                    }
                }
            }
        }
        return lchild;
    }
    
    private static int getIndex(char root,char[] num) {//找根在数组中的位置
        int index=0;
        for(int i=0;i<num.length;i++){
            if(num[i]==root) index=i;
        }
        return index;
    }
}

运行结果