【LeetCode】1234. 替换子串得到平衡字符串
1234. 替换子串得到平衡字符串
题目描述
有一个只含有 ‘Q’, ‘W’, ‘E’, ‘R’ 四种字符,且长度为 n 的字符串。
假如在该字符串中,这四个字符都恰好出现 n/4 次,那么它就是一个「平衡字符串」。
给你一个这样的字符串 s,请通过「替换一个子串」的方式,使原字符串 s 变成一个「平衡字符串」。
你可以用和「待替换子串」长度相同的 任何 其他字符串来完成替换。
请返回待替换子串的最小可能长度。
如果原字符串自身就是一个平衡字符串,则返回 0。
示例 1
输入:s = “QWER”
输出:0
解释:s 已经是平衡的了。
示例 2
输入:s = “QQWE”
输出:1
解释:我们需要把一个 ‘Q’ 替换成 ‘R’,这样得到的 “RQWE” (或 “QRWE”) 是平衡的。
示例 3
输入:s = “QQQW”
输出:2
解释:我们可以把前面的 “QQ” 替换成 “ER”。
提示
- 1 <= s.length <= 105
- s.length 是 4 的倍数
- s 中只含有 ‘Q’, ‘W’, ‘E’, ‘R’ 四种字符
算法一:计数+双指针(滑动窗口)
思路
- 首先使用一个哈希表统计四个字符的出现次数, 「当四个字符的出现次数不超过 s.size() / 4 」时, 不需要替换子串,直接返回 0。
- 否则, 使用双指针 i , j 维护滑动窗口的左右边界, 初始都设置为 0 。
- 接下来,遍历字符串 s, 每次遍历到某字符, 就将它的出现次数减一, 然后判断当前窗口是否满足条件,即窗口外的字符数量都不超过 s.size() / 4 ,如果满足条件,那么就更新答案, 然后就将窗口左移(右边界 i 固定, 增大 j ,所以窗口变小), 直到不满足条件为止。
收获
- 我没有注意到题目要求的是将子串替换,也就是需要将连续个字符替换,因此一开始方向错了;
- 题解用了 t.find(s[i]) ,更方便计数,这个需要学习;
- 此外,还学习了 移动窗口 的思想。
算法情况
- 时间复杂度:O(n),其中 n 为字符串 s 的长度;
- 空间复杂度:O(C),C 为 4。
代码
class Solution {
public:int balancedString(string s) {int n = s.size();int m = n / 4; // 每个字符出现次数string t = "QWER";vector<int> cnt(4);for(auto &c : s){cnt[t.find(c)]++;}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.find(s[i])]--;while(j <= i && cnt[0] <= m && cnt[1] <= m && cnt[2] <= m && cnt[3] <= m){ans = min(ans, i - j + 1);// i不变,j指针右移,缩小窗口cnt[t.find(s[j++])]++;}}return ans;}
};
参考资料:
- string.find()的用法