Leetcode Binary Tree Right Side View

最新推荐文章于 2024-09-09 22:01:06 发布

markpen

最新推荐文章于 2024-09-09 22:01:06 发布

阅读量178

点赞数

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

本文链接：https://blog.csdn.net/markpen/article/details/53891940

版权

leetcode 专栏收录该内容

167 篇文章 0 订阅

订阅专栏

题意：求出树每一层最右边的元素。

思路：BFS，记录每一层的结点，输出最后一个结点。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> re;
    vector<int> rightSideView(TreeNode* root) {
        if(root == NULL) return re;
        vector<TreeNode*> t;
        t.push_back(root);
        bfs(t);
        
        return re;
    }
    
    void bfs(vector<TreeNode*> l) {
        if(l.size() == 0) return;
        
        re.push_back(l[l.size() - 1]->val);
        
        vector<TreeNode*> temp;
        for(int i = 0; i < l.size(); ++ i) {
            if(l[i] == NULL) continue;
            if(l[i]->left != NULL) temp.push_back(l[i]->left);
            if(l[i]->right != NULL) temp.push_back(l[i]->right);
        }
        
        bfs(temp);
        
        return;
        
    }
};