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

输出二叉树的右视图

https://www.nowcoder.com/practice/c9480213597e45f4807880c763ddd5f0

class Solution {
  public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 求二叉树的右视图
     * @param preOrder int整型vector 先序遍历
     * @param inOrder int整型vector 中序遍历
     * @return int整型vector
     */

    int current_index = 0; // 前序遍历数组的索引

    //建树
    TreeNode* reConstructBinaryTree(vector<int>& preOrder, vector<int>& vinOrder) {
        if (preOrder.empty() || vinOrder.empty())
            return nullptr;
        TreeNode* head = new TreeNode(preOrder[current_index]);
        vector<int> newLeftVin, newRightVin;
        int i = 0;
        while (vinOrder[i] != preOrder[current_index]) {
            newLeftVin.push_back(vinOrder[i]);
            ++i;
        }
        ++i;
        while (i != vinOrder.size()) {
            newRightVin.push_back(vinOrder[i]);
            ++i;
        }
        vector<int> newPre;
        current_index++;

        head->left =  reConstructBinaryTree(preOrder, newLeftVin);
        head->right = reConstructBinaryTree(preOrder, newRightVin);

        return head;
    }

    //设置双队列,q1放父节点,q2放子节点,每次把q2队尾加入最后结果即可。放完以后把q1更新成q2,并清空q2
    vector<int> solve(vector<int>& preOrder, vector<int>& inOrder) {
        vector<int> res;

        TreeNode* head = reConstructBinaryTree(preOrder, inOrder);

        if (!head)
            return res;

        queue<TreeNode*> q1, q2;
        q1.push(head);
        int n = q1.back()->val;
        res.push_back(n);
        while (!q1.empty()) {
            while (!q1.empty()) {
                if (q1.front()->left) {
                    q2.push(q1.front()->left);
                }
                if (q1.front()->right) {
                    q2.push(q1.front()->right);
                }
                q1.pop();
            }
            if (!q2.empty()) {
                n = q2.back()->val;
                res.push_back(n);
            }
            while (!q2.empty()) {
                q1.push(q2.front());
                q2.pop();
            }
        }
        return res;
    }
};

方法比较传统,总体就两步:建树+找每层最右节点

建树的思路就不赘述了,请看上一题。找最右节点可以设置双队列,q1放父节点,q2放子节点,每次把q2的队尾,也就是每层的最右节点,加入最后结果即可。放完以后把q1更新成q2,并清空q2

全部评论

相关推荐

不愿透露姓名的神秘牛友
昨天 12:31
以前小时候我最痛恨出轨、偷情的人,无论男女,为什么会出轨?现在我成了自己最讨厌的人,没想到分享的东西在牛客会被这么多人看,大家的评价都很中肯,我也认同,想过一一回复,但我还是收声了,我想我应该说说这件事,这件事一直压在我心里,是个很大的心结,上面说了人为什么出轨,我大概能明白了。我们大一下半年开始恋爱,开始恋爱,我给出了我铭记3年的承诺,我对她好一辈子,我永远不会背叛,我责任心太重,我觉得跟了我,我就要照顾她一辈子,我们在一起3年我都没有碰过她,她说往东我就往东,她说什么我做什么,她要我干什么,我就干什么!在学校很美好,中途也出过一些小插曲,比如男闺蜜、男闺蜜2号等等等。但我都强迫她改掉了,我...
牛客刘北:两个缺爱的人是没有办法好好在一起的,但世界上哪有什么是非对错?你后悔你们在一起了,但是刚刚在一起的美好也是真的呀,因为其他人的出现,你开始想要了最开始的自己,你的确对不起自己,21岁的你望高物远,你完全可以不谈恋爱,去过你想要的生活,你向往自由,在一起之后,你要想的不是一个人,而是两个人,你不是变心了,就像你说的,你受够了,你不想包容了,冷静几天是你最优的选择,爱人先爱己。
社会教会你的第一课
点赞 评论 收藏
分享
06-02 15:53
阳光学院 Java
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务