牛客网刷题-输出二叉树的右视图

问题描述

请根据二叉树的前序遍历，中序遍历恢复二叉树，并打印出二叉树的右视图。

输入描述:
输入二叉树的前序和中序遍历

输出描述:
输出二叉树的右视图

示例

示例1

输入
[1,2,4,5,3],[4,2,5,1,3]

输出
[1,3,5]

解决思路

分析

1. 根据二叉树的前序和中序还原二叉树，然后通过广度优先遍历得到二叉树的右视图

方法

1. 根据二叉树的前序和中序遍历的特点还原二叉树（可以看考这篇文章二叉树的重建）
2. 通过队列实现二叉树的广度优先遍历，得到最右侧的节点，从而得到右视图。

代码实现

// 思路1
public class Solution {
    public int[] solve(int[] pre, int[] in) {
        // write code here
        TreeNode root = recoverTree(pre, in);
        List<Integer> list = getRightView(root);
        int[] res = new int[list.size()];
        for (int i = 0; i < list.size(); i++) {
            res[i] = list.get(i);
        }
        return res;
    }

    //  重构二叉树
    public TreeNode recoverTree(int[] pre, int[] in) {
        if (pre.length == 0 || in.length == 0) {
            return null;
        }
        // 前序第一个节点为跟节点
        TreeNode root = new TreeNode(pre[0]);
        // 遍历中序找跟节点
        for (int i = 0; i < in.length; i++) {
            if (pre[0] == in[i]) {
                root.left = recoverTree(Arrays.copyOfRange(pre, 1, i + 1), Arrays.copyOfRange(in, 0, i));
                root.right = recoverTree(Arrays.copyOfRange(pre, i + 1, pre.length), Arrays.copyOfRange(in, i + 1, in.length));
            }
        }

        return root;
    }

    // 查看右视图，右视图即二叉树从右向左看的视图
    // 采用广度优先搜索遍历，搜索每一层，只保留该层最后一个节点
    public List<Integer> getRightView(TreeNode root) {
        // 通过链表实现队列
        LinkedList<TreeNode> list = new LinkedList<>();
        // 存储右视图
        List<Integer> res = new ArrayList<>();
        // 添加跟节点
        list.add(root);
        while (!list.isEmpty()) {
            // size主要用于记录每一层的节点数，方便跳出循环遍历
            int size = list.size();
            TreeNode cur = null;
            while (size > 0) {
                size--;
                // 从队列的第一个节点开始取（即从左向右遍历获取，最后一个节点为右视图）
                cur = list.removeFirst();
                // 向队列中添加左节点
                if (cur.left != null) {
                    list.add(cur.left);
                }
                // 向队列中添加右节点
                if (cur.right != null) {
                    list.add(cur.right);
                }
            }
            // 节点的最后一个值方法右视图
            res.add(cur.val);
        }
        // 返回右视图
        return res;
    }
}

小伙伴如果想测试的话，可以直接到牛客网这个链接做测试

输出二叉树的右视图-牛客网