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
结果:
复杂度分析:
时间复杂度:O(n),其中 n 是二叉树的节点数。
空间复杂度:O(n),其中 n 是二叉树的节点数。