侧面观察二叉树
题目
给定一棵二叉树,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
示例:
输入: [1,2,3,null,5,null,4,6]
输出: [1, 3, 4, 6]
解释:
思路
提前了解:层次遍历(宽度优先搜索)请点击:二叉树基础知识
在右侧观察到二叉树的数,其实就是二叉树每一层的最右边的一个数,如果需要找到最右边的一个数,可以通过层次遍历去了解每一个节点的层数,然后由层次遍历的顺序去修改该层数下的值,由于层次遍历也是在该层的情况下,从左往右遍历二叉树,所以该层最后一个遍历的数就是最右侧的数。(过程展示如下图)
代码
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) {
vector<int> view;
queue<pair<TreeNode*, int>> Q;
if(root){
Q.push(make_pair(root, 0));
}
while(!Q.empty()){ //层次遍历为框架
TreeNode* node = Q.front().first;
int depth= Q.front().second;
Q.pop();
if(view.size() == depth){ //第一次遍历到该层的数时
view.push_back(node->val);
}
else{ //还在该层遍历时
view[depth] = node->val;
}
if(node->left){
Q.push(make_pair(node->left, depth + 1));
}
if(node->right){
Q.push(make_pair(node->right, depth + 1));
}
}
return view;
}
};
致谢
本章知识点和思路由小象学院相关视频提供,由本人学习并梳理得出,希望自己加深记忆的同时,也能给大家提供更多有关于一些算法的知识点。
你的点赞、评论、收藏就是对我最大的支持与鼓励,谢谢!