第156场双周赛T1~T4 & 第449场周赛T1~T4 (8)

Author: gdut-yy

leetcode/leetcode-36/src/main/java/Solution3541.java

@@ -0,0 +1,39 @@
+public class Solution3541 {
+ // final int VOWEL_MASK = (1 << ('a' - 'a')) | (1 << ('e' - 'a')) | (1 << ('i' - 'a')) | (1 << ('o' - 'a')) | (1 << ('u' - 'a'));
+ private static final int VOWEL_MASK = 1065233;
+
+ public int maxFreqSum(String s) {
+ int[] cnt = new int[26];
+ for (char c : s.toCharArray()) {
+ cnt[c - 'a']++;
+ }
+ int mx1 = 0, mx2 = 0;
+ for (int i = 0; i < 26; i++) {
+ if (isVowel(i)) {
+ mx1 = Math.max(mx1, cnt[i]);
+ } else {
+ mx2 = Math.max(mx2, cnt[i]);
+ }
+ }
+ return mx1 + mx2;
+ }
+
+ private boolean isVowel(int b) {
+ return (VOWEL_MASK >> b & 1) == 0;
+ }
+}
+/*
+3541. 找到频率最高的元音和辅音
+https://leetcode.cn/problems/find-most-frequent-vowel-and-consonant/description/
+
+第 156 场双周赛 T1。
+
+给你一个由小写英文字母（'a' 到 'z'）组成的字符串 s。你的任务是找出出现频率 最高 的元音（'a'、'e'、'i'、'o'、'u' 中的一个）和出现频率最高的辅音（除元音以外的所有字母），并返回这两个频率之和。
+注意：如果有多个元音或辅音具有相同的最高频率，可以任选其中一个。如果字符串中没有元音或没有辅音，则其频率视为 0。
+一个字母 x 的 频率 是它在字符串中出现的次数。
+提示:
+1 <= s.length <= 100
+s 只包含小写英文字母
+
+模拟。
+ */

leetcode/leetcode-36/src/main/java/Solution3542.java

@@ -0,0 +1,59 @@
+import java.util.ArrayDeque;
+import java.util.Deque;
+
+public class Solution3542 {
+ static class V1 {
+ public int minOperations(int[] nums) {
+ int ans = 0;
+ Deque<Integer> st = new ArrayDeque<>();
+ for (int x : nums) {
+ while (!st.isEmpty() && x < st.peek()) {
+ st.pop();
+ ans++;
+ }
+ // 如果 x 与栈顶相同，那么 x 与栈顶可以在同一次操作中都变成 0，x 无需入栈
+ if (st.isEmpty() || x != st.peek()) {
+ st.push(x);
+ }
+ }
+ return ans + st.size() - (st.getLast() == 0 ? 1 : 0); // 0 不需要操作
+ }
+ }
+
+ static class V2 {
+ public int minOperations(int[] nums) {
+ int ans = 0;
+ int top = -1; // 栈顶下标（把 nums 当作栈）
+ for (int x : nums) {
+ while (top >= 0 && x < nums[top]) {
+ top--; // 出栈
+ ans++;
+ }
+ // 如果 x 与栈顶相同，那么 x 与栈顶可以在同一次操作中都变成 0，x 无需入栈
+ if (top < 0 || x != nums[top]) {
+ nums[++top] = x; // 入栈
+ }
+ }
+ return ans + top + (nums[0] > 0 ? 1 : 0);
+ }
+ }
+}
+/*
+3542. 将所有元素变为 0 的最少操作次数
+https://leetcode.cn/problems/minimum-operations-to-convert-all-elements-to-zero/description/
+
+第 156 场双周赛 T2。
+
+给你一个大小为 n 的 非负 整数数组 nums 。你的任务是对该数组执行若干次（可能为 0 次）操作，使得 所有 元素都变为 0。
+在一次操作中，你可以选择一个子数组 [i, j]（其中 0 <= i <= j < n），将该子数组中所有 最小的非负整数 的设为 0。
+返回使整个数组变为 0 所需的最少操作次数。
+一个 子数组 是数组中的一段连续元素。
+提示:
+1 <= n == nums.length <= 10^5
+0 <= nums[i] <= 10^5
+
+赛时没做出来。。
+单调栈。
+https://leetcode.cn/problems/minimum-operations-to-convert-all-elements-to-zero/solutions/3673804/cong-fen-zhi-dao-dan-diao-zhan-jian-ji-x-mzbl/
+rating 1822 (clist.by)
+ */

leetcode/leetcode-36/src/main/java/Solution3543.java

@@ -0,0 +1,74 @@
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.HashSet;
+import java.util.List;
+import java.util.Set;
+
+public class Solution3543 {
+ private List<int[]>[] g;
+ private int k, t, ans;
+ private Set<String> memo;
+
+ public int maxWeight(int n, int[][] edges, int k, int t) {
+ g = new ArrayList[n];
+ Arrays.setAll(g, e -> new ArrayList<>());
+ for (int[] e : edges) {
+ g[e[0]].add(new int[]{e[1], e[2]}); // y, wt
+ }
+ this.k = k;
+ this.t = t;
+
+ memo = new HashSet<>();
+ ans = -1;
+ for (int x = 0; x < n; x++) {
+ dfs(x, 0, 0);
+ }
+ return ans;
+ }
+
+ private void dfs(int x, int i, int s) {
+ String key = x + ":" + i + ":" + s;
+ if (!memo.add(key)) return;
+ if (i == k) {
+ ans = Math.max(ans, s);
+ return;
+ }
+ for (int[] p : g[x]) {
+ int y = p[0], wt = p[1];
+ if (s + wt < t) {
+ dfs(y, i + 1, s + wt);
+ }
+ }
+ }
+}
+/*
+3543. K 条边路径的最大边权和
+https://leetcode.cn/problems/maximum-weighted-k-edge-path/description/
+
+第 156 场双周赛 T3。
+
+给你一个整数 n 和一个包含 n 个节点（编号从 0 到 n - 1）的 有向无环图（DAG）。该图由二维数组 edges 表示，其中 edges[i] = [ui, vi, wi] 表示一条从节点 ui 到 vi 的有向边，边的权值为 wi。
+同时给你两个整数 k 和 t。
+你的任务是确定在图中边权和 尽可能大的 路径，该路径需满足以下两个条件：
+- 路径包含 恰好 k 条边；
+- 路径上的边权值之和 严格小于 t。
+返回满足条件的一个路径的 最大 边权和。如果不存在这样的路径，则返回 -1。
+提示:
+1 <= n <= 300
+0 <= edges.length <= 300
+edges[i] = [ui, vi, wi]
+0 <= ui, vi < n
+ui != vi
+1 <= wi <= 10
+0 <= k <= 300
+1 <= t <= 600
+输入图是 有向无环图（DAG）。
+不存在重复的边。
+
+DFS
+https://leetcode.cn/problems/maximum-weighted-k-edge-path/solutions/3673824/liang-chong-fang-fa-dfs-tuo-bu-xu-dppyth-vnva/
+时间复杂度 O((n+m)kt)。其中 m 是 edges 的长度。
+相似题目: 3509. 最大化交错和为 K 的子序列乘积
+https://leetcode.cn/problems/maximum-product-of-subsequences-with-an-alternating-sum-equal-to-k/description/
+rating 2092 (clist.by)
+ */

leetcode/leetcode-36/src/main/java/Solution3544.java

@@ -0,0 +1,84 @@
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+
+public class Solution3544 {
+ private List<Integer>[] g;
+ private int[] nums;
+ private int k;
+ private long[][][] memo;
+
+ public long subtreeInversionSum(int[][] edges, int[] nums, int k) {
+ int n = nums.length;
+ this.nums = nums;
+ this.k = k;
+ g = new ArrayList[n];
+ Arrays.setAll(g, e -> new ArrayList<>());
+ for (int[] e : edges) {
+ g[e[0]].add(e[1]);
+ g[e[1]].add(e[0]);
+ }
+ memo = new long[n][k + 1][2];
+ for (int i = 0; i < n; i++) {
+ for (int j = 0; j < k + 1; j++) {
+ memo[i][j][0] = Long.MIN_VALUE;
+ memo[i][j][1] = Long.MAX_VALUE;
+ }
+ }
+ return dfs(0, -1, k)[0];
+ }
+
+ private long[] dfs(int x, int fa, int d) {
+ long mx = nums[x], mn = nums[x];
+ if (memo[x][d][0] != Long.MIN_VALUE) return memo[x][d];
+ for (Integer y : g[x]) {
+ if (y == fa) continue;
+ long[] sub = dfs(y, x, Math.min(d + 1, k));
+ mx += sub[0];
+ mn += sub[1];
+ }
+ if (d >= k) {
+ long mx2 = -nums[x], mn2 = -nums[x];
+ for (Integer y : g[x]) {
+ if (y == fa) continue;
+ long[] sub = dfs(y, x, 1);
+ mx2 -= sub[1];
+ mn2 -= sub[0];
+ }
+ mx = Math.max(mx, mx2);
+ mn = Math.min(mn, mn2);
+ }
+ return memo[x][d] = new long[]{mx, mn};
+ }
+}
+/*
+3544. 子树反转和
+https://leetcode.cn/problems/subtree-inversion-sum/description/
+
+第 156 场双周赛 T4。
+
+给你一棵以节点 0 为根节点包含 n 个节点的无向树，节点编号从 0 到 n - 1。该树由长度为 n - 1 的二维整数数组 edges 表示，其中 edges[i] = [ui, vi] 表示节点 ui 和 vi 之间有一条边。
+同时给你一个整数 k 和长度为 n 的整数数组 nums，其中 nums[i] 表示节点 i 的值。
+你可以对部分节点执行 反转操作 ，该操作需满足以下条件：
+- 子树反转操作：
+ - 当你反转一个节点时，以该节点为根的子树中所有节点的值都乘以 -1。
+- 反转之间的距离限制：
+ - 你只能在一个节点与其他已反转节点“足够远”的情况下反转它。
+ - 具体而言，如果你反转两个节点 a 和 b，并且其中一个是另一个的祖先（即 LCA(a, b) = a 或 LCA(a, b) = b），那么它们之间的距离（它们之间路径上的边数）必须至少为 k。
+返回应用 反转操作 后树上节点值的 最大可能 总和 。
+在一棵有根树中，某个节点 v 的子树是指所有路径到根节点包含 v 的节点集合。
+提示:
+2 <= n <= 5 * 10^4
+edges.length == n - 1
+edges[i] = [ui, vi]
+0 <= ui, vi < n
+nums.length == n
+-5 * 10^4 <= nums[i] <= 5 * 10^4
+1 <= k <= 50
+输入保证 edges 表示的是一棵合法的树。
+
+树形 DP
+时间复杂度 O(nk)。
+有 O(n) 的做法。
+rating 2551 (clist.by)
+ */

leetcode/leetcode-36/src/main/java/Solution3545.java

@@ -0,0 +1,35 @@
+import java.util.Arrays;
+
+public class Solution3545 {
+ public int minDeletion(String s, int k) {
+ int[] cnt = new int[26];
+ for (char c : s.toCharArray()) {
+ cnt[c - 'a']++;
+ }
+ Arrays.sort(cnt);
+
+ int ans = s.length();
+ for (int i = 0; i < k; i++) {
+ int j = 25 - i;
+ ans -= cnt[j];
+ }
+ return ans;
+ }
+}
+/*
+3545. 不同字符数量最多为 K 时的最少删除数
+https://leetcode.cn/problems/minimum-deletions-for-at-most-k-distinct-characters/description/
+
+第 449 场周赛 T1。
+
+给你一个字符串 s（由小写英文字母组成）和一个整数 k。
+你的任务是删除字符串中的一些字符（可以不删除任何字符），使得结果字符串中的 不同字符数量 最多为 k。
+返回为达到上述目标所需删除的 最小 字符数量。
+提示：
+1 <= s.length <= 16
+1 <= k <= 16
+s 仅由小写英文字母组成。
+
+贪心。
+时间复杂度 O(ClogC)。其中 C = 26。
+ */

leetcode/leetcode-36/src/main/java/Solution3546.java

@@ -0,0 +1,51 @@
+public class Solution3546 {
+ private long[][] ps2d;
+
+ public boolean canPartitionGrid(int[][] grid) {
+ int m = grid.length;
+ int n = grid[0].length;
+
+ ps2d = new long[m + 1][n + 1];
+ for (int i = 1; i <= m; i++) {
+ for (int j = 1; j <= n; j++) {
+ ps2d[i][j] = ps2d[i - 1][j] + ps2d[i][j - 1] - ps2d[i - 1][j - 1] + grid[i - 1][j - 1];
+ }
+ }
+ if (ps2d[m][n] % 2 != 0) return false;
+ long target = ps2d[m][n] / 2;
+
+ for (int i = 0; i < m; i++) {
+ if (sumRegion(0, 0, i, n - 1) == target) return true;
+ }
+ for (int j = 0; j < n; j++) {
+ if (sumRegion(0, 0, m - 1, j) == target) return true;
+ }
+ return false;
+ }
+
+ // 求 [r1,c1] 到 [r2,c2] 的累加和
+ private long sumRegion(int r1, int c1, int r2, int c2) {
+ return ps2d[r2 + 1][c2 + 1] - ps2d[r2 + 1][c1] - ps2d[r1][c2 + 1] + ps2d[r1][c1];
+ }
+}
+/*
+3546. 等和矩阵分割 I
+https://leetcode.cn/problems/equal-sum-grid-partition-i/description/
+
+第 449 场周赛 T2。
+
+给你一个由正整数组成的 m x n 矩阵 grid。你的任务是判断是否可以通过 一条水平或一条垂直分割线 将矩阵分割成两部分，使得：
+- 分割后形成的每个部分都是 非空 的。
+- 两个部分中所有元素的和 相等 。
+如果存在这样的分割，返回 true；否则，返回 false。
+提示：
+1 <= m == grid.length <= 10^5
+1 <= n == grid[i].length <= 10^5
+2 <= m * n <= 10^5
+1 <= grid[i][j] <= 10^5
+
+二维前缀和 + 枚举。
+时间复杂度 O(mn)。
+相似题目: 3548. 等和矩阵分割 II
+https://leetcode.cn/problems/equal-sum-grid-partition-ii/description/
+ */