LeetCode Binary Tree Right Side View

最新推荐文章于 2020-04-18 21:00:57 发布

pantingting_

最新推荐文章于 2020-04-18 21:00:57 发布

阅读量290

点赞数 1

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.csdn.net/pantingting_zju/article/details/50807652

版权

leetcode 专栏收录该内容

25 篇文章 0 订阅

订阅专栏

层次遍历

我的想法：层序遍历，然后输出每一层最右边的节点代码如下：

class Solution {
public:
    vector<int> rightSideView(TreeNode* root) {
        vector<int> result;
        vector<TreeNode*> high;
        vector<TreeNode*> low;
        if(root==NULL) return result;
        high.push_back(root);
        while(high.size()!=0){
            result.push_back(high[high.size()-1]->val);
            for(int i=0;i<high.size();i++){
                if(high[i]->left!=NULL) low.push_back(high[i]->left);
                if(high[i]->right!=NULL) low.push_back(high[i]->right);
            }
            high=low;
            low.clear();
        }
        return result;
    }
};

先序遍历

参考讨论区的回答后，采用了改进的先序遍历来做，遍历顺序为根-》右子树-》左子树。当一个节点被遍历到的时候，如果结果集中的节点数比当前的level值小的话，说明当前节点是本层中第一个被遍历到的，因为我们先遍历右子树，所以当前节点一定是本层中最右节点。代码如下

class Solution {
public:
    void preOrder(TreeNode* root,int level,vector<int>&res){
        if(root==NULL) return;
        if(level>res.size()) res.push_back(root->val);
        preOrder(root->right,level+1,res);
        preOrder(root->left,level+1,res);
    }
    vector<int> rightSideView(TreeNode* root) {
        vector<int> res;
        preOrder(root,1,res);
        return res;
    }
};