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”。

示例 4：

输入：s = “QQQQ”
输出：3
解释：我们可以替换后 3 个 ‘Q’，使 s = “QWER”。

提示：

1 <= s.length <= 10^5
s.length 是 4 的倍数
s 中只含有 'Q', 'W', 'E', 'R' 四种字符

题解：一开始看题目有点儿懵，动态规划？怎么找状态转移方程？后来静心下来思考发现不是，这样理解，题目是要求你找出一段子区间，然后子区间内容你随意操作，使得结果的字符串是平衡的（也就是4种字符Q,W,E,R的数目都是n/4），那么我们会发现一个趣点，就是比如字符串长度为16，也就是每个字符应该有4个才能平衡，假设Q有14个，那么假设我在选择一个子区间后，除去这个子区间中的Q，其他部分还有5个Q，那么这个子区间是不可取的，这个可以理解吧，因为你可以把这个子区间的内容变成Q增加Q的数目，但是没法减少Q的数目，所以对选择的子区间只有一个要求：**选择这个区间后，除去这个区间中的Q,W,E,R，剩余部分的Q,W,E,R数目必须都小于等于n/4。**那么是不是可以利用前缀和快速的查找某个区间中的Q,W,E,R的数目，于是我们可以利用二分查找，找区间[l,r]，l可以从左边0一直到右边，r利用二分查找，可以肯定的是，r越大，区间越大包含的内容越多，肯定符合要求，并且二分查找时间复杂度为nlogn应该不会超时，但是我突然想到，都可以这样二分查找了，肯定满足双指针查询的数据属性要求啊，废话不说直接写了双指针，就通过了，美滋滋。

AC代码

class Solution {
public:
int Q[100100],W[100100],E[100100],R[100100];
int balancedString(string s) {
    int q=0,w=0,e=0,r=0;
    memset(Q,0,sizeof(Q));
    memset(W,0,sizeof(W));
    memset(E,0,sizeof(E));
    memset(R,0,sizeof(R));
    for(int i=0;i<s.length();i++)
    {
        if(s[i]=='Q')
        {
            q++;
            Q[i+1]=Q[i]+1;
        }
        else Q[i+1]=Q[i];
        if(s[i]=='W')
        {
            w++;
            W[i+1]=W[i]+1;
        }
        else W[i+1]=W[i];
        if(s[i]=='E')
        {
            e++;
            E[i+1]=E[i]+1;
        }
        else E[i+1]=E[i];
        if(s[i]=='R')
        {
            r++;
            R[i+1]=R[i]+1;
        }
        else R[i+1]=R[i];
    }
    int st=0,et=1,ansq,answ,anse,ansr;
    int n=s.length();
    int m=n/4;
    int mi=1e9;
    while(st<n&&et<n)
    {
        ansq=Q[et+1]-Q[st];
        answ=W[et+1]-W[st];
        anse=E[et+1]-E[st];
        ansr=R[et+1]-R[st];
        if((q-ansq)<=m&&(w-answ)<=m&&(e-anse)<=m&&(r-ansr)<=m)判断是否满足条件
        {
            mi=min(mi,et-st+1);
            st++;
        }
        else et++;
    }
    return mi;
}
};

效果还不错哦~