题解 | #输出二叉树的右视图#

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     * 求二叉树的右视图
     * @param xianxu int整型一维数组 先序遍历
     * @param zhongxu int整型一维数组 中序遍历
     * @return int整型一维数组
     */
    public void helper(int[] xianxu, int[] zhongxu, Map<Integer,List<Integer>> map,int level){
        if(xianxu.length==0) return;
        int now = xianxu[0];
            int idx = 0;
        for(;idx<zhongxu.length;idx++){
            if(zhongxu[idx]==now) break;
        }
        if(map.containsKey(level)) map.get(level).add(now);
            else {
                map.put(level, new ArrayList());
                map.get(level).add(now);
            }
        helper(Arrays.copyOfRange(xianxu,1,idx+1),Arrays.copyOfRange(zhongxu,0,idx),map,level+1);
        helper(Arrays.copyOfRange(xianxu,idx+1,xianxu.length),Arrays.copyOfRange(zhongxu,idx+1,zhongxu.length),map,level+1);
        
    }
    public int[] solve (int[] xianxu, int[] zhongxu) {
        // write code here
        Map<Integer,List<Integer>> map = new HashMap();
        helper(xianxu, zhongxu, map, 0);
        Set kset = map.keySet();
        int[] res = new int[kset.size()];
        for(int i=0;i<kset.size();i++){
                List<Integer> temp = map.get(i);
                res[i] = temp.get(temp.size()-1);
            }
        return  res;
    }
}