NC136—输出二叉树的右视图

最新推荐文章于 2023-03-06 09:17:01 发布

fkuner

最新推荐文章于 2023-03-06 09:17:01 发布

阅读量287

点赞数

CC 4.0 BY-SA版权

分类专栏： NC

本文链接：https://blog.csdn.net/younothings/article/details/110263066

NC 专栏收录该内容

26 篇文章

订阅专栏

本文介绍了一种根据二叉树的前序遍历和中序遍历恢复二叉树的方法，并通过广度优先搜索算法实现了二叉树右视图的打印。使用了哈希表来快速查找节点的位置。

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

题意

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

题解

#include <unordered_map>
class Solution {
private:
    unordered_map<int, int> findIndex;
    int index;
    vector<int> res;
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     * 求二叉树的右视图
     * @param xianxu int整型vector 先序遍历
     * @param zhongxu int整型vector 中序遍历
     * @return int整型vector
     */
    vector<int> solve(vector<int>& xianxu, vector<int>& zhongxu) {
        this->index = 0;
        int len = zhongxu.size();
        for(int i = 0; i < len; i++){
            findIndex[zhongxu[i]] = i;
        }
        TreeNode* root = build(xianxu, 0, len-1);
        bfs(root);
        return res;
    }
    
    TreeNode* build(vector<int>& preOrder, int start, int end){
        if(start > end) return nullptr;
        int mid = findIndex[preOrder[index]];
        TreeNode* root = new TreeNode(preOrder[index++]);
        root->left = build(preOrder, start, mid-1);
        root->right = build(preOrder, mid+1, end);
        return root;
    }
    
    void bfs(TreeNode* root){
        queue<TreeNode*> q;
        q.push(root);
        while(!q.empty()){
            int len = q.size();
            for(int i = 0; i < len; i++){
                TreeNode* node = q.front();
                if(node->left != nullptr)
                    q.push(node->left);
                if(node->right != nullptr)
                    q.push(node->right);
                if(i == len - 1){
                    res.push_back(node->val);
                }
                q.pop();
            }
        }
    }
};