Leetcode199. 二叉树的右视图

Every day a Leetcode

题目来源:199. 二叉树的右视图

解法1:层序遍历

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

按层序遍历,将每层的最后一个节点值放入答案数组中。

代码:

/*
 * @lc app=leetcode.cn id=199 lang=cpp
 *
 * [199] 二叉树的右视图
 */

// @lc code=start
/**
 * Definition for a binary tree node.
 * 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)
    {
        // 特判
        if (root == nullptr)
            return {};
        vector<int> ans;
        queue<TreeNode *> q;
        q.push(root);
        while (!q.empty())
        {
            int size = q.size();
            for (int i = 0; i < size; i++)
            {
                auto node = q.front();
                q.pop();
                if (i == size - 1)
                    ans.push_back(node->val);
                if (node->left)
                    q.push(node->left);
                if (node->right)
                    q.push(node->right);
            }
        }
        return ans;
    }
};
// @lc code=end

结果:

Leetcode199. 二叉树的右视图_第1张图片

复杂度分析:

时间复杂度:O(n),其中 n 是二叉树的节点数。

空间复杂度:O(n),其中 n 是二叉树的节点数。

你可能感兴趣的:(Every,day,a,leetcode,leetcode,C++,算法,数据结构与算法)