199. 二叉树的右视图

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);
        }
    }
};
posted @ 2019-10-18 20:33  NeoZy  阅读(131)  评论(0编辑  收藏  举报