Leetcode - 132双周赛

目录

一、3174. 清除数字

二、3175. 找到连续赢 K 场比赛的第一位玩家

三、3176. 求出最长好子序列 I

四、3177. 求出最长好子序列 II

---

一、3174. 清除数字

本题可以使用栈来模拟，遇到数字弹出栈顶元素，遇到字母入栈。

代码如下：

//使用字符串模拟栈实现
class Solution {
    public String clearDigits(String s) {
        StringBuilder st = new StringBuilder();
        for(char c : s.toCharArray()){
            if(Character.isDigit(c)){
                st.deleteCharAt(st.length()-1);
            }else{
                st.append(c);
            }
        }
        return st.toString();
    }
}

二、3175. 找到连续赢 K 场比赛的第一位玩家

 本题可以使用双端队列模拟，但是要注意当 k >= n 时，直接返回最大值

代码如下：

class Solution {
    public int findWinningPlayer(int[] s, int k) {
        int n = s.length;
        int mx = 0, idx = 0;
        Deque<int[]> que = new ArrayDeque<>();
        for(int i=0; i<n; i++){
            que.addLast(new int[]{s[i], i});
            if(mx < s[i]){
                mx = s[i];
                idx = i;
            }
        }
        if(k >= n) return idx;
        int t = k;
        mx = que.poll()[0];
        idx = 0;
        while(t > 0){
            int[] x = que.removeFirst();
            if(x[0] > mx){
                que.addLast(new int[]{mx, idx});
                mx = x[0];
                idx = x[1];
                t = k-1;
            }else{
                que.addLast(x);
                t--;
            }
        }
        return idx;
    }
}

我们发现当循环大于 n 的时候，返回的结果一定是 max(skills)，所以我们只需要考虑在一个循环内能否找出连赢 k 场的玩家，可以使用一个变量 cnt 记录当前玩家的胜利次数，如果 cnt >= k，直接返回答案。否则返回 max(skills)

代码如下： 

class Solution {
    public int findWinningPlayer(int[] skills, int k) {
        int idx = 0;
        int cnt = 0;
        for(int i=1; i<skills.length; i++){
            if(skills[i] > skills[idx]){
                idx = i;
                cnt = 0;
            }
            cnt++;
            if(cnt >= k) return idx;
        }
        return idx;
    }
}

三、3176. 求出最长好子序列 I

本题可以使用dfs枚举选哪个/选或不选来解决，这里讲解枚举选哪个的做法。先定义dfs，我们需要当前下标 i，根据题目子序列中 seq[i] != seq[i+1] 的数不能超过 k 个，我们还需要上一次选择的下标 j，以及 seq[i] != seq[i+1] 的次数 k。所以可以定义 dfs(i，j，k)：前一个数为nums[j]时，[i，n-1]的中取出最大子序列，且该子序列 seq[i] != seq[i+1] 的次数不超过 k。

但是这么写记忆化时需要三个参数，会导致时间复杂度过大，如何优化呢？可以发现上面是从前往后遍历，所以枚举时需要一个参数来告诉它起始点是什么。但是如果反过来遍历，就可以直接省去这个参数，因为我们是从 j - 1，往前遍历，重新定义 dfs(j，k)：后一个数为nums[j]时，[0，j-1]的中取出最大子序列，且该子序列 seq[i] != seq[i+1] 的次数不超过 k。

代码如下： 

class Solution {
    public int maximumLength(int[] nums, int k) {
        int n = nums.length;
        memo = new int[n+1][k+1];
        for(int[] row : memo)
            Arrays.fill(row, -1);
        return dfs(n,k,nums);
    }
    int[][] memo;
    int dfs(int j, int k, int[] nums){
        int res = 0;
        if(memo[j][k] != -1) return memo[j][k]; 
        for(int i=0; i<j; i++){
            if(j==nums.length || nums[i] == nums[j])
                res = Math.max(res, dfs(i, k, nums)+1);
            else if(k > 0)
                res = Math.max(res, dfs(i, k-1, nums)+1);
        }
        return memo[j][k] = res;
    }
}

递推写法

f[i][j]：以 i 为结尾的子序列中最多有 j 个满足相邻元素不相等的最长子序列长度

假设 k < i：

• nums[i] == nums[k]，f[i][j] = Math.max(f[i][j]，f[k][j]+1)
• nums[i] != nums[k]，f[i][j] = Math.max(f[i][j]，f[k][j]+1)

class Solution {
    public int maximumLength(int[] nums, int k) {
        int n = nums.length;
        int[][] f = new int[n+1][k+1];
        int ans = 0;
        Map<Integer, Integer> map = new HashMap<>();
        for(int i = 0; i < n; i++){
            f[i][0] = map.merge(nums[i], 1, Integer::sum);
            ans = Math.max(ans, f[i][0]);
        }
        for(int j = 1; j <= k; j++){
            f[0][j] = f[0][0];
        }
        for(int j=1; j<n; j++){
            for(int a=1; a<=k; a++){ 
                for(int i=0; i<j; i++){    
                    if(nums[i]==nums[j])
                        f[j][a] = Math.max(f[j][a], f[i][a]+1);
                    else
                        f[j][a] = Math.max(f[j][a], f[i][a-1]+1);
                }
            }
            ans = Math.max(ans, f[j][k]);
        }
        return ans;
    }
}

 

四、3177. 求出最长好子序列 II

本题和上一题一样，但是数据范围更大，上述n^2*k的做法超时了，如何优化?？我们可以把f[i][j] 中的 i 转换成对应的 nums[i]，这时 f[x][j]：以 x (即nums[i]）结尾的子序列中最多有 j 个满足相邻元素不相等的最长子序列长度。

假设 k < i：

• nums[k] == x，f[x][j] = Math.max(f[x][j]，f[x][j]+1)
• nums[k] != x，f[x][j] = Math.max(f[x][j]，f[nums[k]][j-1]+1)

上述写法可以合并 f[x][j] = Math.max(f[x][j]+1，f[nums[k]][j-1]+1)，我们可以额外维护一个mx[j] = max(f[nums[k]][j-1])就可以在O(1)时间算出f[x][j]

public class Solution {
    public int maximumLength(int[] nums, int k) {
        Map<Integer, int[]> fs = new HashMap<>();
        int[] mx = new int[k + 2];
        for (int x : nums) {
            int[] f = fs.computeIfAbsent(x, i -> new int[k + 1]);
            for (int j = k; j >= 0; j--) {
                f[j] = Math.max(f[j], mx[j]) + 1;
                mx[j + 1] = Math.max(mx[j + 1], f[j]);//同01背包，防止覆盖还需使用的值
            }
        }
        return mx[k + 1];
    }
}