给定一棵二叉树,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
示例:
输入: [1,2,3,null,5,null,4]
输出: [1, 3, 4]
解释:
1 <---
/ \
2 3 <---
\ \
5 4 <---
通过次数48,613提交次数76,288
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-right-side-view
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
二叉树的层次遍历,利用队列对二叉树进行层次遍历,队列中不仅存储二叉树节点,同时存储当前节点的深度,这时候我们就能够获得深度变化,即每一层最右边的节点,值得注意的是,当队列为空的时候,之前pop出去的节点也应该是当前层最右边的节点。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<int> rightSideView(TreeNode* root) {
queue<pair<TreeNode*,int>> q;
vector<int> ret;
if(root == nullptr)return ret;
int deepth = 0;
q.push(pair<TreeNode*,int>(root,deepth));
int count = 0;
while(!q.empty())
{
TreeNode* temp = q.front().first;
int tempDeepth = q.front().second;
q.pop();
if(q.empty()||tempDeepth!=q.front().second)ret.push_back(temp->val);
if(temp->left)
{
q.push(pair<TreeNode*,int>(temp->left,tempDeepth+1));
}
if(temp->right)
{
q.push(pair<TreeNode*,int>(temp->right,tempDeepth+1));
}
}
return ret;
}
};