给定一棵二叉树,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
示例:
输入: [1,2,3,null,5,null,4]
输出: [1, 3, 4]
解释:
1 <—
/
2 3 <—
\
5 4 <—
两种方法,bfs和dfs
bfs:
//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> result;
if(root==NULL) return result;
queue<TreeNode*> q;
q.push(root);
while(!q.empty()){
int size = q.size();
for(int i=0;i<size;i++){
TreeNode *node = q.front();
q.pop(); //删掉第一个元素
if(node->left!=NULL) q.push(node->left);
if(node->right!=NULL) q.push(node->right);
if(i==size-1) result.push_back(node->val);
}
}
return result;
}
};
dfs:
class Solution {
public:
vector<int> rightSideView(TreeNode* root) {
vector<int> res;
dfs(root,0,res);
return res;
}
/*当res大小和level大小相等时,push_back当前的元素,而因为又是从右开始访问,
所以正好是右边第一个元素
*/
void dfs(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);
}
};