力扣 199题二叉树的右视图记录

楚洋的小跟班

于 2024-07-22 17:30:08 发布

阅读量151

点赞数

分类专栏：二叉树文章标签： leetcode 算法 c++

本文链接：https://blog.csdn.net/m0_49800270/article/details/140615276

版权

二叉树专栏收录该内容

6 篇文章 0 订阅

订阅专栏

题目描述

给定一个二叉树的 根节点 root，想象自己站在它的右侧，按照从顶部到底部的顺序，返回从右侧所能看到的节点值。

示例 1:
输入: [1,2,3,null,5,null,4]
输出: [1,3,4]

示例 2:
输入: [1,null,3]
输出: [1,3]

示例 3:
输入: []
输出: []

思路

层次遍历，当遍历到每一层的最后的一个节点的时候，将该节点的值放入 result 数组中。

完整代码

#include<iostream>
#include<queue>
#include<vector>
#include<algorithm>

using namespace std;

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode() : val(0), left(nullptr), right(nullptr) {}
    TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
    TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
};

class Solution {
public:
    vector<int> rightSideView(TreeNode* root) {
        vector<int> result;
        queue<TreeNode*> que;
        if(root != NULL) que.push(root);
        while(!que.empty()){
            int size = que.size();
            for(int i = 0; i < size; i++){
                TreeNode* node = que.front();
                que.pop();
                if(i == size - 1) result.push_back(node->val);
                if(node->left) que.push(node->left);
                if(node->right) que.push(node->right); 
            }
        }
        return result;
    }
};

int main()
{
    TreeNode* root = new TreeNode(1);
    root->left = new TreeNode(2);
    root->left->right = new TreeNode(5);
    root->right = new TreeNode(3);
    root->right->right= new TreeNode(4);

    Solution s;
    vector<int> result = s.rightSideView(root);

    for(int val : result){
        cout << val << " "; 
    }
    cout << endl;

    return 0;
}