算法笔记
本题可以在层序遍历的基础上进行小小的改动就可以完成,只要掌握了层序遍历,解题起来就会相对轻松不少。
题目描述
解题步骤
本题要求返回从右侧能看到的节点值,那是不是就可以理解为返回层序遍历每一层中最后一个节点,那么就可以很容易想到使用层序遍历去解题,那么对于另一个问题,如何返回该层的最后一个节点?
我们知道我们每次在层序遍历代码实现中使用size去记录定义队列的长度,结合层序遍历的逻辑,弹出父节点,那么它们的孩子进入队列,那么当我们for循环当中的i等于size - 1的时候,就相当于弹出了每一层的最后一个节点,即需要返回的节点,那么这道题就可以迎刃而解。
代码部分
class Solution {
public:
vector<int> rightSideView(TreeNode* root) {
vector<int> result;
queue<TreeNode*> que;
if(root != nullptr) {
que.push(root);
}
while(!que.empty()) {
int size = que.size();
for(int i = 0; i < size; i++) {
TreeNode* node = que.front();
que.pop();
if(i == (size - 1)) {
result.push_back(node -> val);
}
if(node -> left) {
que.push(node -> left);
}
if(node -> right) {
que.push(node -> right);
}
}
}
return result;
}
};
结语
本题作为使用层序遍历解决的第三个问题,主要还是加深对于层序遍历总体逻辑的理解,后面仍会继续对于层序遍历相关题目进行解析。