【每日一题】【BFS&Lambda&重建二叉树】2022年2月15日-根据先序中序重建并输出二叉树的右视图...

最新推荐文章于 2025-08-17 21:38:20 发布

哥们要飞

最新推荐文章于 2025-08-17 21:38:20 发布

阅读量124

点赞数

CC 4.0 BY-SA版权

文章标签：宽度优先 java 算法开发语言

本文链接：https://blog.csdn.net/USTSD/article/details/125518177

该博客主要介绍了如何根据二叉树的前序遍历和中序遍历恢复二叉树，并实现层次遍历以获取二叉树的右视图。通过递归的重构方法建立二叉树结构，然后利用队列进行层次遍历，在遍历过程中记录每一层的最后一个节点，即为右视图的节点值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

描述

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

思路：重建&层次遍历记录最后一个&Lambda表达式

答案：

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     * 求二叉树的右视图
     * @param xianxu int整型一维数组 先序遍历
     * @param zhongxu int整型一维数组 中序遍历
     * @return int整型一维数组
     */
    //思路：使用队列BFS层次遍历，当到达最后一个节点时加入res
    public int[] solve (int[] xianxu, int[] zhongxu) {
        List<Integer> res = new ArrayList<>();
        //先确定二叉树
        TreeNode root = reconstrution(xianxu, 0, xianxu.length - 1, zhongxu, 0, zhongxu.length - 1);
        //使用队列，对树进行层次遍历
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while(!queue.isEmpty()) {
            int size = queue.size();
            for(int i = 0; i < size; i++) {
                if(i == size - 1) {
                    res.add(queue.peek().val);
                }
                TreeNode node = queue.poll();
                if(node.left != null) {
                    queue.offer(node.left);
                }
                if(node.right != null) {
                    queue.offer(node.right);
                }
            }
        }
        return res.stream().mapToInt(x -> x).toArray();
    }
    //递归，类似回溯：路径、选择列表、结束条件
    //但是回溯要移除选择，而递归无需移除选择
    public TreeNode reconstrution(int[] xianxu, int preStart, int preEnd, int[] zhongxu, int inStart, int inEnd) {
        if(preStart > preEnd || inStart > inEnd) {
            return null;
        }
        TreeNode root = new TreeNode(xianxu[preStart]);
        //找到中序的位置
        int i = 0;
        for(i = inStart; i <= inEnd; i++) {
            if(zhongxu[i] == xianxu[preStart]) {
                break;
            }
        }
        root.left = reconstrution(xianxu, preStart + 1, preStart + (i - inStart), zhongxu, inStart, i - 1);
        root.right = reconstrution(xianxu, preStart + (i - inStart) + 1, preEnd, zhongxu, i + 1, inEnd);
        return root;
    }
}