2022-10-30 第 317 场周赛

6220. 可被三整除的偶数的平均值

class Solution {
    public int averageValue(int[] nums) {
        int sum = 0, times = 0;
        for (int num : nums) {
            if (num % 2 == 0 && num % 3 == 0) {
                sum += num;
                times++;
            }
        }
        return times == 0 ? 0 : sum / times;
    }
}

6221. 最流行的视频创作者

class Solution {
    public List<List<String>> mostPopularCreator(String[] creators, String[] ids, int[] views) {
        int n = ids.length;
        Map<String,long[]> map = new HashMap<>();
        long max = Long.MIN_VALUE;
        for (int i = 0; i < n; i++) {
            String c = creators[i];
            if (map.containsKey(creators[i])) {
                map.get(c)[1] += views[i];
                if (views[(int)map.get(c)[0]] < (long)views[i]) {
                    map.get(c)[0] = i;
                } else if (views[(int)map.get(c)[0]] == (long)views[i]) {
                    map.get(c)[0] = ids[(int)map.get(c)[0]].compareTo(ids[i]) <= 0 ? map.get(c)[0] : i;
                }
            } else {
                map.put(c,new long[]{i,views[i]});
            }
            max = Math.max(max,map.get(c)[1]);
        }
        List<List<String>> ans = new ArrayList<>();
        for (String str : map.keySet()) {
            if (map.get(str)[1] == max) {
                List<String> list = new ArrayList<>();
                list.add(str);
                list.add(ids[(int)map.get(str)[0]]);
                ans.add(list);
            }
        }
        return ans;
    }
}

6222. 美丽整数的最小增量

分析

贪心：从个位开始一位一位的向左添加数使得对应数位变为0，减少数位和，直到满足条件。

class Solution {
    public long makeIntegerBeautiful(long n, int target) {
        long m = n, i = 10;
        while (true) {
            int sum = 0;
            for (long j = m; j > 0; j = j / 10) {
                sum += j % 10;
            }
            if (sum <= target) {
                return m - n;
            }
            m = m / i * i + i;
            i *= 10;
        }
    }
}

6223. 移除子树后的二叉树高度

分析

计算出删除每个节点之后，整棵树的深度是多少。
删除一个子树的根节点，树的高度有可能是三种情况：
1.除了被删除节点父节点以外的树的最大深度
2.被删除节点的兄弟节点所能到达的最大深度
3.被删除节点父节点这个位置的深度

两次DFS，第一次计算出每个节点所能到达的最大深度，用map存储下来。
第二次DFS是为了计算第一种情况：除了被删除节点父节点以外的深度。
当前节点被删除之后剩余树的高度是有它的上一层产生的。

class Solution {
    Map<TreeNode,Integer> map = new HashMap<>();
    Map<Integer,Integer> mapRest = new HashMap<>();
    public int[] treeQueries(TreeNode root, int[] queries) {
        dfs(root,0);
        fillMap(root,0,0);
        int[] ans = new int[queries.length];
        for (int i = 0; i < queries.length; i++) {
            ans[i] = mapRest.get(queries[i]);
        }
        return ans;
    }

    public int dfs(TreeNode root, int level) {
        if (root == null) {
            return level - 1;
        }
        int max = Math.max(dfs(root.left,level+1),dfs(root.right,level+1));
        map.put(root,max);
        return max;
    }

    public void fillMap(TreeNode root, int rest, int level) {
        if (root == null) {
            return;
        }
        mapRest.put(root.val,rest);
        fillMap(root.left,Math.max(rest,Math.max(level,map.getOrDefault(root.right,0))),level+1);
        fillMap(root.right,Math.max(rest,Math.max(level,map.getOrDefault(root.left,0))), 1+level);
    }
}