题目描述
给定一棵二叉树,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
输入: [1,2,3,null,5,null,4]
输出: [1, 3, 4]
解释:
1 <---
/ \
2 3 <---
\ \
5 4 <---
解题思路
这是一道层次遍历的变形,常规题。我的实现。
- 广度优先搜索:层次遍历的变形。
- 深度优先搜索:按“中-右-左”顺序遍历二叉树,在
dfs
函数中多加一个level
参数,在res.size() == level
时才res.push_back(root->val)
,即表示每一层只保存最右边的那个节点。
参考代码
bfs
/**
* 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) {
vector<int> res;
if(root == nullptr)
return res;
queue<TreeNode*> q;
q.push(root);
while(!q.empty()){
res.push_back(q.front()->val);
int num = q.size();
while(num--){
TreeNode* pNode = q.front();
q.pop();
if(pNode->right) q.push(pNode->right);
if(pNode->left) q.push(pNode->left);
}
}
return res;
}
};
dfs
class Solution {
public:
vector<int> rightSideView(TreeNode* root) {
vector<int> res;
dfs(root, 0, res);
return res;
}
void helper(TreeNode* root, int level, vector<int>& res){
if(!root) return;
if(res.size() == level)
res.push_back(root->val);
dfs(root->right, level+1, res);
dfs(root->left, level+1, res);
}
};