[杨小白]_leetcode_第311场周赛-第三题6182. 反转二叉树的奇数层

112 阅读2分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第2天,点击查看活动详情

前言

小白算法比较菜,希望能激励我每日更新,从leetcode第一题开始,2022年目标300题,记录从0到1的全过程!!

第311场周赛-力扣

第311场周赛-力扣

这场周赛好像挺简单的,但是第四题超时没做出来。1小时18分三题已经排到接近4000名了。掉大分!!

前两题没什么好说的了,第四题不太会,就讲讲第三题吧。

自己做的算法可真是复杂,看别人写的可真简单,,,,,,

6182. 反转二叉树的奇数层

给你一棵 完美 二叉树的根节点 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 。

image.png

示例 2

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

image.png

示例 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 是一棵 完美 二叉树

2.解法一

我自己想的办法确实太复杂,思路不算难,先用了个arraylist里面装了很多个list,对树进行层序遍历。

然后再根据层的序号,如果是奇数就反向遍历,重新构建树。确实太复杂了。

class Solution {
        ArrayList<ArrayDeque<Integer>> arr = new ArrayList<>();

        public TreeNode reverseOddLevels(TreeNode root) {
            ArrayDeque<TreeNode> deque = new ArrayDeque<>();
            deque.add(root);
            ArrayDeque<Integer> integers = new ArrayDeque<>();
            integers.add(root.val);
            arr.add(integers);
            while (!deque.isEmpty()) {
                ArrayDeque<TreeNode> deque2 = new ArrayDeque<>();
                integers = new ArrayDeque<>();
                while (!deque.isEmpty()) {
                    TreeNode poll = deque.poll();
                    if (poll == null) continue;
                    if (poll.left != null){
                        deque2.add(poll.left);
                        integers.add(poll.left.val);
                    }
                    if (poll.right != null) {
                        deque2.add(poll.right);
                        integers.add(poll.right.val);
                    }
                }
                deque.addAll(deque2);
                if (!integers.isEmpty()) {
                    arr.add(integers);
                }
            }
            // System.out.println(arr);
            TreeNode res = new TreeNode(arr.get(0).getFirst());
            ArrayDeque<TreeNode> up = new ArrayDeque<>();
            up.add(res);
            pre(1,up);
            return res;

        }

        public void pre(int depth,ArrayDeque<TreeNode> up) {
            if (depth==arr.size()) return;
            ArrayDeque<Integer> down = arr.get(depth);
            ArrayDeque<TreeNode> up2 = new ArrayDeque<>();
            if (depth % 2 == 0) {
                while (!up.isEmpty()) {
                    TreeNode treeNode = up.removeFirst();
                    treeNode.left = new TreeNode(down.removeFirst());
                    treeNode.right = new TreeNode(down.removeFirst());
                    up2.add(treeNode.left);
                    up2.add(treeNode.right);
                }
            } else {
                while (!up.isEmpty()) {
                    TreeNode treeNode = up.removeFirst();
                    treeNode.left = new TreeNode(down.removeLast());
                    treeNode.right = new TreeNode(down.removeLast());
                    up2.add(treeNode.left);
                    up2.add(treeNode.right);
                }
            }
            pre(depth+1,up2);
        }
    }

image.png

2.解法二

dfs方法。

这个方法是直接将两个节点的val进行对调,真是太巧妙了~

class Solution {
    public void dfs(TreeNode root1, TreeNode root2, int depth) {
        if(root1==null) return;
        if(depth%2 == 1) {
            int temp = root1.val;
            root1.val = root2.val;
            root2.val = temp;
        }
        dfs(root1.left, root2.right, depth + 1);
        dfs(root1.right, root2.left, depth + 1);
    }
    public TreeNode reverseOddLevels(TreeNode root) {
        dfs(root.left, root.right, 1);
        return root;
    }
}

image.png

3.结束

三题选手都4000名了,而且掉分了,掉大分了,真是哭了,下次一定下次一定,gogogo,刷题刷题,每天一道,三年1000道!!!!