QUESTION
medium
题目描述
给定一棵二叉树,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值
示例:
输入: [1,2,3,null,5,null,4]
输出: [1, 3, 4]
解释:
1 <---
/ \
2 3 <---
\ \
5 4 <---
说明
无
SOLUTION
看这个题目是要求优先遍历右子树,但是要注意不能简单的直接往下优先只遍历右节点,左节点也是同样需要的
下面这种情况如果你仅仅遍历了右节点,那你可能得不到正确结果
输入: [1,2,3,null,4]
输出: [1, 3, 4]
解释:
1 <---
/ \
2 3 <---
\
4 <---
方法一
回到问题,其实最容易想到的办法应该是层次遍历输出每层最右边的节点,那只要在层次遍历的基础上修改一下就可以了
class Solution {
public:
vector<int> rightSideView(TreeNode* root) {
vector<int> res;
if(!root) return res;
queue<TreeNode*> q;
q.push(root);
while(!q.empty()){
for(int i = q.size(); i > 0; --i){
TreeNode* t = q.front();
q.pop();
if(t->left) q.push(t->left);
if(t->right) q.push(t->right);
if(i == 1) res.push_back(t->val);
}
}
return res;
}
};
方法二
先序右遍历(我这么称呼的,可能有正式名称)
- 先序遍历是:根->左->右;这个遍历是:根->右->左
- 注意符合条件时的判断方法
class Solution {
public:
vector<int> rightSideView(TreeNode* root) {
vector<int> res;
preRightOrder(root, res, 0);
return res;
}
void preRightOrder(TreeNode* node, vector<int> &res, int deep){
if(!node) return;
if(deep == res.size()) res.push_back(node->val);
preRightOrder(node->right, res, deep + 1);
preRightOrder(node->left, res, deep + 1);
}
};
当然这种方法应该也有非递归的办法吧,没试过,大家可以自己试试