Q:
给定一棵二叉树,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
示例:
输入: [1,2,3,null,5,null,4]
输出: [1, 3, 4]
解释:
1 <—
/
2 3 <—
\
5 4 <—
A:
解法1:常规层次遍历每次取最后一个元素:
class Solution {
public:
vector<int> rightSideView(TreeNode* root)
{
queue<TreeNode*> myqueue;
myqueue.push(root);
TreeNode* cur=0;
vector<int> res;
if(!root)
{
return res;
}
while(!myqueue.empty())
{
TreeNode* my_back=myqueue.back();
res.push_back(my_back->val);
while(cur!=my_back)
{
cur=myqueue.front();
myqueue.pop();
if(cur->left)
{
myqueue.push(cur->left);
}
if(cur->right)
{
myqueue.push(cur->right);
}
}
}
return res;
}
};
解法2:利用一个depth变量记录当前考察节点的层数,对该节点来说下一层先考察右子节点,再考察左子节点。通过比较结果序列的长度和层数的大小,如果同一层中右侧节点已经放入结果序列(即序列长度大于等于层数),该层的左侧节点就不会再放入序列。
class Solution {
public:
vector<int> rightSideView(TreeNode* root)
{
int depth=0;
vector<int>res;
func(res,root,1);
return res;
}
void func(vector<int>& res,TreeNode* node,int depth)
{
if(node==nullptr)
{
return;
}
if(res.size()>=depth) //该层已经放置元素
{
func(res,node->right,depth+1);
func(res,node->left,depth+1);
}
else
{
res.push_back(node->val);
func(res,node->right,depth+1);
func(res,node->left,depth+1);
}
}
};