Leetcode 199 Binary Tree Right Side View 二叉树右视图

时间:2022-11-16 14:36:53

原题地址

https://leetcode.com/problems/binary-tree-right-side-view/

题目描述

Given a binary tree, imagine yourself standing on the right side of it, return the values of the nodes you can see ordered from top to bottom.
给出一个二叉树,设想你从树的右侧看过来,返回你从上往下可以看到的结点.

For example:
例如:

Given the following binary tree,
给出如下的二叉树

   1            <---
 /   \
2     3         <---
 \     \
  5     4       <---

You should return [1, 3, 4].
你需要返回[1, 3, 4].

解题思路

首先可以明确的是,看到的结点是每一层的结点中最右侧的结点,我们要做的就是按照层序把每一层最右侧的结点找出来并返回.然后针对按照层序的关键点,我们可以考虑层序遍历的类似思路.层序遍历中需要用到队列,这里我们考虑也使用队列.但是与层序遍历不同的是,我们需要非常明确的区分开不同的层,以保存每层的最后一个结点.我们可以使用两个队列,一个队列中存储某一层的结点,另一个队列存储下一层的结点,这样做到把不同层的结点区分开来.

代码 cpp

class Solution {
public:
    /** 获取每一层最右侧的结点 */
    vector<int> rightSideView(TreeNode* root) {
        /* 两个队列交替使用,获取队尾的元素 */
        vector<int> ret;
        queue<TreeNode*> queues[2];

        /* 初始化,把第一层结点放入第一个队列*/
        if (root != NULL) queues[0].push(root);

        /* 交替使用队列,一个队列存储一层结点 * 另一个队列存储下一层结点(或上一层结点) */
        int i = 0, j = 1, tmp;
        TreeNode *p;        
        while (!queues[0].empty() || !queues[1].empty()) {
            while (!queues[i].empty()) {
                p = queues[i].front();
                queues[i].pop();
                if (p->left) queues[j].push(p->left);
                if (p->right) queues[j].push(p->right);
                tmp = p->val;
            }
            // 遍历完某一层结点时,tmp是该层最右侧的结点的值
            ret.push_back(tmp); // 添加到结果集
            i = (i + 1) % 2; // 交换两个队列的角色
            j = (j + 1) % 2;
        }

        return ret;
    }
};

完整代码 https://github.com/Orange1991/leetcode/blob/master/199/cpp/main.cpp


2015/8/5