1.题目
给定一棵二叉树,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
示例:
输入: [1,2,3,null,5,null,4]
输出: [1, 3, 4]
解释:
1 <---
/ \
2 3 <---
\ \
5 4 <---
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-right-side-view
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2.题解
DFS
可以使用dfs搜索,用depth变量来记录深度,每层从右子节点开始搜索,每层第一个遇到的就是你可以看到的那个节点。
class Solution
{
public:
vector<int> rightSideView(TreeNode* root)
{
vector<int> ans;
if(root==NULL) return ans;
dfs(root,0,ans);
return ans;
}
void dfs(TreeNode* root,int depth,vector<int>& ans)
{
if(root==NULL) return;
if(depth==ans.size()) ans.push_back(root->val);
dfs(root->right,depth+1,ans);
dfs(root->left,depth+1,ans);
}
};
BFS
与常规bfs相比就是每次遍历右子树节点先入队,每次保存最右边的节点。
class Solution
{
public:
vector<int> rightSideView(TreeNode* root)
{
vector<int> res;
if(!root) return res;
queue<TreeNode*> q;
q.push(root);
while(!q.empty())
{
res.push_back(q.back()->val);
int size=q.size();
for(int i=0; i<size; ++i)
{
TreeNode* t=q.front();
q.pop();
if(t->left) q.push(t->left);
if(t->right) q.push(t->right);
}
}
return res;
}
};