当前位置：首页 > news >正文

LeetCode 题目「二叉树的右视图」中，如何从「中间存储」到「一步到位」实现代码的优化？

news 来源：原创 2025/5/23 21:40:17

背景简介

在 LeetCode 的经典题目 「二叉树的右视图」 中，我们需要返回从右侧看一棵二叉树时所能看到的节点集合。每一层我们只能看到最右边的那个节点。

最初，我采用了一个常规思路：层序遍历 + 每层单独保存节点值 + 最后提取每层最后一个节点。但在逐步分析中，我发现这一过程可以进一步优化，减少中间变量的使用，提高代码简洁度和可读性。

初始版本思路（注释中的方式）

下面是我最初的设计思路片段：

while(!que.empty()){
    int size = que.size();
    vector<int> vec; // 用于存储当前层所有节点的值

    for(int i = 0; i < size; i++){
        TreeNode* node = que.front();
        que.pop();
        vec.push_back(node -> val); // 保存该节点值
        if(node -> left) que.push(node -> left);
        if(node -> right) que.push(node -> right);
    }

    res.push_back(vec[size - 1]); // 每层最后一个加入结果
}

✅ 优点：

清晰直观
完全符合“层序遍历 + 每层最后一个”这一概念

❌ 缺点：

使用了额外的 vector<int> vec 存储每层节点值，造成冗余
实际我们并不需要保留整层的所有值，只关心最后一个

优化思路

我们只需要每层的最后一个节点值，那么在遍历这一层时，只需判断当前访问的是不是“最后一个节点”。如果是，就将其值加入结果集即可，无需保留整层节点值。

于是我们这样优化：

while(!que.empty()){
    int size = que.size();

    for(int i = 0; i < size; i++){
        TreeNode* node = que.front();
        que.pop();
        
        // 只记录这一层的最后一个节点
        if(i == size - 1) res.push_back(node -> val);
        
        if(node -> left) que.push(node -> left);
        if(node -> right) que.push(node -> right);
    }
}

最终代码（优化后）

class Solution {
public:
    vector<int> rightSideView(TreeNode* root) {
        vector<int> res;
        queue<TreeNode*> que;
        if(root) 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) res.push_back(node -> val);
                if(node -> left) que.push(node -> left);
                if(node -> right) que.push(node -> right);
            }
        }
        return res;
    }
};