LC.2415 | 反转二叉树的奇数层 | 树| 两次BFS (记录+回填)

输入:一棵 完美二叉树 的根节点 root

要求:反转这棵树中每个 奇数 层(第 1, 3, 5... 层)的节点值 注意:根节点在第 0 层(偶数层),不需要反转

输出:反转操作后的二叉树根节点


思路:很典型的层序遍历题,不用多想,两次遍历稳稳搞定。

  1. 第一遍 BFS(数据采集与处理): 按层遍历整棵树。 将每一层的节点值收集到一个临时列表 current_layer_vals 中。关键点:如果当前层是 奇数层,直接在存入大数组 levels 之前,将这个列表进行 反转 (reverse)。这样 levels 里存的就是这一层最终该有的顺序。

  2. 第二遍 BFS(数据回填): 再次按层遍历整棵树。 如果当前遍历到了 奇数层,就从之前准备好的 levels 数组中取出对应的数值进行赋值。


复杂度:

        时间复杂度:O(N)

        空间复杂度:O(N)


class Solution {
public:
    TreeNode* reverseOddLevels(TreeNode* root) {
        if (!root) return nullptr;

        vector<vector<int>> levels;
        queue<TreeNode*> q;
        q.push(root);

        while (!q.empty()) {
            int size = q.size();
            vector<int> current_layer_vals;
            
            for (int i = 0; i < size; i++) {
                TreeNode* node = q.front();
                q.pop();
                current_layer_vals.push_back(node->val);
                
                if (node->left) q.push(node->left);
                if (node->right) q.push(node->right);
            }
            
            if (levels.size() % 2 != 0) {
                reverse(current_layer_vals.begin(), current_layer_vals.end());
            }
            
            levels.push_back(current_layer_vals);
        }

        q.push(root);
        int level_index = 0;

        while (!q.empty()) {
            int size = q.size();
            
            for (int i = 0; i < size; i++) {
                TreeNode* node = q.front();
                q.pop();

                if (level_index % 2 != 0) {
                    node->val = levels[level_index][i];
                }
                
                if (node->left) q.push(node->left);
                if (node->right) q.push(node->right);
            }
            level_index++;
        }

        return root;
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值