图解LeetCode第 199 号问题:二叉树的右视图

来自:五分钟学算法(微信号:blgczzz),作者:程序员小吴

该文已加入开源项目:LeetCodeAnimation(用动画的形式呈现解LeetCode题目的思路,目前 5500 Star )。地址:https://github.com/MisterBooo/LeetCodeAnimation

LeetCode上第 199 号问题:二叉树的右视图

题目描述

给定一棵二叉树,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。

示例:

输入: [1,2,3,null,5,null,4]    
输出: [1, 3, 4]   
解释:

思路解析

与之前二叉树的层次遍历类似的,该问题需要用到队列,

  • 建立一个queue

  • 遍历每层的节点时,把下一层的节点都存入到queue中

  • 每当开始新一层节点的遍历之前,先把新一层最后一个节点值存到结果中

动画演示

动画演示GIF加载有点慢,请稍等片刻^_^

参考代码

 1class Solution {
2public:
3    vector<int> rightSideView(TreeNode *root) {
4        vector<int> res;
5        if (!root) return res;
6        queue<TreeNode*> q;
7        q.push(root);
8        while (!q.empty()) {
9            res.push_back(q.back()->val);
10            int size = q.size();
11            for (int i = 0; i < size; ++i) {
12                TreeNode *node = q.front();
13                q.pop();
14                if (node->left) q.push(node->left);
15                if (node->right) q.push(node->right);
16            }
17        }
18        return res;
19    }
20};

代码截图

推荐↓↓↓
算法与数据结构
上一篇:看动画轻松理解「链表」实现「LRU缓存淘汰算法」 下一篇:看动画轻松理解「 堆 」