[杨小白]_leetcode_998. 最大二叉树 II

66 阅读2分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第33天,点击查看活动详情

前言

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

8月30日每日一题

998. 最大二叉树 II

最大树 定义:一棵树,并满足:其中每个节点的值都大于其子树中的任何其他值。

给你最大树的根节点 root 和一个整数 val 。

就像 之前的问题 那样,给定的树是利用 Construct(a) 例程从列表 a(root = Construct(a))递归地构建的:

如果 a 为空,返回 null 。 否则,令 a[i] 作为 a 的最大元素。创建一个值为 a[i] 的根节点 root 。 root 的左子树将被构建为 Construct([a[0], a[1], ..., a[i - 1]]) 。 root 的右子树将被构建为 Construct([a[i + 1], a[i + 2], ..., a[a.length - 1]]) 。 返回 root 。 请注意,题目没有直接给出 a ,只是给出一个根节点 root = Construct(a) 。

假设 b 是 a 的副本,并在末尾附加值 val。题目数据保证 b 中的值互不相同。

返回 Construct(b) 。

示例 1

image.png

输入:root = [4,1,3,null,null,2], val = 5

输出:[5,4,null,1,3,null,null,2]

解释:a = [1,4,2,3], b = [1,4,2,3,5]

示例 2

image.png

输入:root = [5,2,4,null,1], val = 3

输出:[5,2,4,null,1,null,3]

解释:a = [2,1,5,4], b = [2,1,5,4,3]

示例 3

image.png

输入:root = [5,2,3,null,1], val = 4

输出:[5,2,4,null,1,3]

解释:a = [2,1,5,3], b = [2,1,5,3,4]

提示:

  • 树中节点数目在范围 [1, 100] 内
  • 1 <= Node.val <= 100
  • 树中的所有值 互不相同
  • 1 <= val <= 100

2.解析

评论区好多同学评论读不懂题,看完一篇确实没看懂题目要求是什么,但是仔细观察每个示例的解释,就能看懂题目问的要求是什么了。

root = [5,2,3,null,1], val = 4

解析如下:题目给了一个树的根节点root,和一个值val

1.将root中序遍历后,得到序列a = [2,1,5,3]

2.将val添加入a的最后,得到b = [2,1,5,3,4]

3.将b进行树的构建,最大值为root.val,最大值左边所有数得到的树为root.left,最大值右边所有数得到的树为root.right。

4.返回得到的树,就可以了

这里的主函数十分简单,rrayList<Integer> arrays = treeToArray(root);这里我写了一个treeToArray函数,可以将树返回中序遍历的数组。

然后再arrays后添加val

最后将此时的arrays重构为树,返回即可。

class Solution {
        public TreeNode insertIntoMaxTree(TreeNode root, int val) {
            ArrayList<Integer> arrays = treeToArray(root);
            arrays.add(val);
            TreeNode res = arrayToTree(arrays, 0, arrays.size() - 1);
            return res;
        }
        public TreeNode arrayToTree(ArrayList<Integer> arrayList, int l, int r) {
            if(l > r) return null;
            int max = arrayList.get(l);
            int maxindex = l;
            for(int i = l; i <= r; i++) {
                if(arrayList.get(i) > max) {
                    max = arrayList.get(i);
                    maxindex = i;
                }
            }
            TreeNode res = new TreeNode(arrayList.get(maxindex));
            res.left = arrayToTree(arrayList,l , maxindex - 1);
            res.right = arrayToTree(arrayList,maxindex + 1, r);
            return res;
        }
        public ArrayList<Integer> treeToArray(TreeNode root) {
            ArrayList<Integer> res = new ArrayList<>();
            if(root==null) return null;
            if(root.left!=null) {
                res.addAll(treeToArray(root.left));
            }
            res.add(root.val);
            if(root.right!=null) {
                res.addAll(treeToArray(root.right));
            }
            return res;
        }
    }

image.png

看时间应该算的上是最优解了

3.结束

![S~P3GBO[X84}YCN`{EN)]U.png

8月31天每日一题打卡完毕,马上就要开学了,加油!!!!!!!!gogogo,刷题刷题,每天一道,三年1000道!!!!