题目概述
一个长度为 n 的字符串 s,其中仅包含 ‘Q’, ‘W’, ‘E’, ‘R’ 四种字符。
如果四种字符在字符串中出现次数均为 n/4,则其为一个平衡字符串。
现可以将 s 中连续的一段子串替换成相同长度的只包含那四个字符的任意字符串,使其变为一个平衡字符串,问替换子串的最小长度?
如果 s 已经平衡则输出0。
输入输出
Input
一行字符表示给定的字符串s
Output
一个整数表示答案
Sample Input
QWER
QQWE
QQQW
QQQQ
Sample Output
0
1
2
3
思路分析
此题用尺取法,首先要明确使用尺取法的条件:
1.所求答案为一个连续区间
2.区间左右端点移动有明确方向
该题符合条件。
首先,遍历字符数组,计算每个字符出现次数,如果相同,则输出0并返回。否则,当右指针r未遍历到字符数组尽头且l<=r时进入while循环。
我们定义sum11-sum44分别用于存储字符串[l,len-1]中除了[l,r]部分的字符QWER的个数。因此要遍历区间[l,r],减去该区间内字符的个数。然后求得sum11-sum44中的最大值,由于我们可以修改的区间为[l,r],先通过替换使4类字符数量一致,再判断剩余空闲位置是否为4的倍数。
如果为4 的倍数,则区间合法,更新答案,l++;否则,r++。
代码
#include<iostream>
#include<cstring>
using namespace std;
const int maxn=1e5+5;
char s[maxn];
int sum1,sum2,sum3,sum4;
int l=0,r=0,ans=100005;
int main()
{
cin>>s;
sum1=0,sum2=0,sum3=0,sum4=0;
int len=strlen(s);
for(int i=0;i<len;i++)
{
if(s[i]=='Q') sum1++;
else if(s[i]=='W') sum2++;
else if(s[i]=='E') sum3++;
else if(s[i]=='R') sum4++;
}
if(sum1==sum2&&sum2==sum3&&sum3==sum4)
{
cout<<0<<endl;
return 0;
}
//sum1-sum4分别记录除[l,r]之外[l,len-1]四种字母的个数
while(r<len&&l<=r){
int sum11=sum1,sum22=sum2,sum33=sum3,sum44=sum4;
for(int i=l;i<=r;i++)
{//除去[l,r]中字符的个数
if(s[i]=='Q') sum11--;
else if(s[i]=='W') sum22--;
else if(s[i]=='E') sum33--;
else if(s[i]=='R') sum44--;
}
int maxx=max(max(sum11,sum22),max(sum33,sum44));
int total=r-l+1;//可修改的位置,即空位
//下面求出将四个字母平衡后剩余的位置
total-=(maxx-sum11)+(maxx-sum22)+(maxx-sum33)+(maxx-sum44);
if(total>=0&&total%4==0)
{//空位为0或4的倍数,区间合法
ans=min(ans,r-l+1);
if(l==r){
l++;
r++;
}
else l++;
}
else r++;
}
cout<<ans<<endl;
return 0;
}
注意
在进行减去区间[l,r]中字符个数操作时,不能直接用
sum1–,
sum2–,
sum3–,
sum4–!因为这样下次进入while循环后sum1-sum4的值就变了!因此需要重新定义一组变量sum11-sum44来记录sum1-sum4并进行–操作。