输入:一棵 完美二叉树 的根节点 root
要求:反转这棵树中每个 奇数 层(第 1, 3, 5... 层)的节点值 注意:根节点在第 0 层(偶数层),不需要反转
输出:反转操作后的二叉树根节点
思路:很典型的层序遍历题,不用多想,两次遍历稳稳搞定。
-
第一遍 BFS(数据采集与处理): 按层遍历整棵树。 将每一层的节点值收集到一个临时列表 current_layer_vals 中。关键点:如果当前层是 奇数层,直接在存入大数组 levels 之前,将这个列表进行 反转 (reverse)。这样 levels 里存的就是这一层最终该有的顺序。
-
第二遍 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;
}
};

被折叠的 条评论
为什么被折叠?



