leetcode 1234. 替换子串得到平衡字符串

226 阅读2分钟

1. 题目与解析

有一个只含有 'Q', 'W', 'E', 'R' 四种字符,且长度为 n 的字符串。

假如在该字符串中,这四个字符都恰好出现 n/4 次,那么它就是一个「平衡字符串」。

 

给你一个这样的字符串 s,请通过「替换一个子串」的方式,使原字符串 s 变成一个「平衡字符串」。

你可以用和「待替换子串」长度相同的 任何 其他字符串来完成替换。

请返回待替换子串的最小可能长度。

如果原字符串自身就是一个平衡字符串,则返回 0

输入: s = "QWER"

输出: 0

解释: s 已经是平衡的了。

输入: s = "QQWE"

输出: 1

解释: 我们需要把一个 'Q' 替换成 'R',这样得到的 "RQWE" (或 "QRWE") 是平衡的。

输入: s = "QQQW"

输出: 2

解释: 我们可以把前面的 "QQ" 替换成 "ER"。

输入: s = "QQQQ"

输出: 3

解释: 我们可以替换后 3 个 'Q',使 s = "QWER"。

针对本题目,我们先用一个哈希表或数组 cnt 统计字符串 sss 中每个字符的数量,如果所有字符的数量都不超过 n/4,那么字符串 sss 就是平衡字符串,直接返回 000。

否则,我们使用双指针 ji 分别维护窗口的左右边界。

接下来,从左到右遍历字符串 s,每次遍历到一个字符,就将该字符的数量减 1,然后判断当前窗口是否满足条件,即窗口外的字符数量都不超过 n/4。如果满足条件,那么就更新答案,然后将窗口的左边界右移,直到不满足条件为止。

2. 题解

class Solution {
    public int balancedString(String s) {
        int[] cnt = new int[4];
        String t = "QWER";
        int n = s.length();
        for (int i = 0; i < n; ++i) {
            cnt[t.indexOf(s.charAt(i))]++;
        }
        int m = n / 4;
        if (cnt[0] == m && cnt[1] == m && cnt[2] == m && cnt[3] == m) {
            return 0;
        }
        int ans = n;
        for (int i = 0, j = 0; i < n; ++i) {
            cnt[t.indexOf(s.charAt(i))]--;
            while (j <= i && cnt[0] <= m 
            && cnt[1] <= m && cnt[2] <= m && cnt[3] <= m) {
                ans = Math.min(ans, i - j + 1);
                cnt[t.indexOf(s.charAt(j++))]++;
            }
        }
        return ans;
    }
}