数据结构与算法之-反转二叉树的奇数层

反转二叉树的奇数层

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

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

反转后，返回树的根节点。

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

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

示例 1：

输入：root = [2,3,5,8,13,21,34]
输出：[2,5,3,8,13,21,34]
解释：
这棵树只有一个奇数层。
在第 1 层的节点分别是 3、5 ，反转后为 5、3 。

示例 2：

输入：root = [7,13,11]
输出：[7,11,13]
解释：
在第 1 层的节点分别是 13、11 ，反转后为 11、13 。

示例 3：

输入：root = [0,1,2,0,0,0,0,1,1,1,1,2,2,2,2]
输出：[0,2,1,0,0,0,0,2,2,2,2,1,1,1,1]
解释：奇数层由非零值组成。
在第 1 层的节点分别是 1、2 ，反转后为 2、1 。
在第 3 层的节点分别是 1、1、1、1、2、2、2、2 ，反转后为 2、2、2、2、1、1、1、1 。

提示：

• 树中的节点数目在范围 [1, 214] 内
• 0 <= Node.val <= 105
• root 是一棵 完美 二叉树

思路：

1. 第一次遍历树时通过HashMap记录奇数层的结点值队列（Deque）。
2. 第二次遍历时从HashMap中取出对应奇数层的Deque，从队尾弹出结点值并更新树的结点。

代码：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    
    Map<Integer, Deque<Integer>> map;
    public TreeNode reverseOddLevels(TreeNode root) {
        map = new HashMap<>();
        find(root, 0);
        change(root, 0);
        return root;
    }
  	// 通过HashMap和Deque记录奇数层结点值
    public void find(TreeNode node, int depth){
        if (node == null) return;
        if (depth % 2 == 1){
            Deque<Integer> q = map.getOrDefault(depth, new LinkedList<>());
            q.offerLast(node.val);
            map.put(depth, q);
        }
        find(node.left, depth + 1);
        find(node.right, depth + 1);
    }
    
  	// 更新结点值
    public void update(TreeNode node, int depth){
        if (node == null) return;
        if (depth % 2 == 1){
            Deque<Integer> q = map.get(depth);
            node.val = q.pollLast();
        }
        update(node.left, depth + 1);
        update(node.right, depth + 1);
    }
    

}