0%

2415-反转二叉树的奇数层

题目

题目链接: https://leetcode.cn/problems/reverse-odd-levels-of-binary-tree/

给你一棵 完美 二叉树的根节点 root ,请你反转这棵树中每个 奇数 层的节点值。

例如,假设第 3 层的节点值是 [2,1,3,4,7,11,29,18] ,那么反转后它应该变成 [18,29,11,7,4,3,1,2] 。
反转后,返回树的根节点。

完美 二叉树需满足:二叉树的所有父节点都有两个子节点,且所有叶子节点都在同一层。

节点的 层数 等于该节点到根节点之间的边数。

思路

遍历当前二叉树,遍历的过程中把每个节点都存储到数组中,然后根据当前层数判断是否需要反转,如果需要反转,则反转当前层数即可。

需要明确一点,给定的二叉树是完美二叉树,所以每一层都是满的,所以每一层都是偶数个节点。

答案

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
/**
* Definition for a binary tree node.
* function TreeNode(val, left, right) {
* this.val = (val===undefined ? 0 : val)
* this.left = (left===undefined ? null : left)
* this.right = (right===undefined ? null : right)
* }
*/
/**
* @param {TreeNode} root
* @return {TreeNode}
*/
var reverseOddLevels = function(root) {
let q = [root]
let isOdd = false
while(q.length) {
if(isOdd) {
const n = q.length
for(let i = 0; i < n / 2; i ++) {
[q[i].val, q[n-1-i].val] = [q[n-1-i].val, q[i].val]
}
}
const tmp = [...q]
q = []
for (const node of tmp) {
if(node.left) {
q.push(node.left)
}
if(node.right) {
q.push(node.right)
}
}
isOdd ^= true
}
return root
};