输出二叉树的右视图

牛客地址

实现代码

import java.util.*;
class TreeNode{
    int val;
    public TreeNode(int val){
        this.val = val;
    }
    TreeNode left;
    TreeNode right;
}

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     * 求二叉树的右视图
     * @param xianxu int整型一维数组 先序遍历
     * @param zhongxu int整型一维数组 中序遍历
     * @return int整型一维数组
     */
    public int[] solve (int[] xianxu, int[] zhongxu) {
        if(xianxu == null || xianxu.length == 0){
            return null;
        }
        // write code here
        ArrayList<Integer> list = new ArrayList<>();
        TreeNode root =  solveCore(xianxu, 0, zhongxu, 0, xianxu.length);
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while(!queue.isEmpty()){
            int size = queue.size();
            for(int i = 0; i <= size - 1; i++){
                TreeNode node = queue.poll();
                if(null != node.left){
                    queue.offer(node.left);
                }
                if(null != node.right){
                    queue.offer(node.right);
                }
                if(i == size - 1){
                    list.add(node.val);
                }
            }
        }
        int size = list.size();
        int[] res = new int[size];
        for(int i = 0; i < size; i++){
            res[i] = list.get(i);
        }
        return res;
    }
    public TreeNode solveCore(int[] preOrder, int preStart, int[] inOrder, int inStart, int len){
        if(len <= 0){
            return null;
        }
        /*if(len == 1){
            return new TreeNode(preOrder[preStart]);
        }*/
        int rootValue = preOrder[preStart];
        TreeNode root = new TreeNode(rootValue);
        int index = inStart;
        while(index < (inStart + len) && inOrder[index] != rootValue){
            index ++;
        }
       // while(index < (inStart + len) && inOrder[index++] != rootValue);
        int leftTreeSize = index - inStart;
        int rightTreeSize = len - 1 - leftTreeSize;
        
        TreeNode left = solveCore(preOrder, preStart + 1, inOrder, inStart, leftTreeSize);
        TreeNode right = solveCore(preOrder, preStart + leftTreeSize + 1, inOrder, index + 1, rightTreeSize);
        root.left = left;
        root.right = right;
        return root;
    }
}

 

 

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值