D2. RGB Substring (hard version)
题意
给你一个长度为N的字符串,求改变最少的字符数量,使N的k个连续字符等于RGBRGBRGB…的子串。
思路
这一题的数据范围是2e5,只能是O(n)或者是O(nlogn)的算法,不能像D1那样暴力做。
这个题需要预处理出来三个字符串,每个字符串都是RGB串,只不过先后顺序不同,然后用给出的字符串与这三个字符串进行比较,如果有相同的贡献就加一,如果不相同贡献就为0。
然后求这三个字符串贡献的前缀和,求出区间为K的最大贡献。
代码
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=2e5+10;
char s[N];
int f1[N],f2[N],f3[N];
int main()
{
int t;
cin>>t;
while(t--)
{
int n,k;
cin>>n>>k;
for(int i=1 ; i<=n+5 ; i++)
{
f1[i]=0;
f2[i]=0;
f3[i]=0;
}
int cnt=0;
for(int i=1 ; i<=n ; i++)
{
cin>>s[i];
cnt++;
if(cnt==1)
{
if(s[i]=='R') f1[i]++;
if(s[i]=='G') f2[i]++;
if(s[i]=='B') f3[i]++;
}
if(cnt==2)
{
if(s[i]=='G') f1[i]++;
if(s[i]=='B') f2[i]++;
if(s[i]=='R') f3[i]++;
}
if(cnt==3)
{
if(s[i]=='B') f1[i]++;
if(s[i]=='R') f2[i]++;
if(s[i]=='G') f3[i]++;
cnt=0;
}
}
/* for(int i=1 ; i<=n ; i++)
{
cout<<f1[i]<<" ";
}
cout<<endl;
for(int i=1 ; i<=n ; i++)
{
cout<<f2[i]<<" ";
}
cout<<endl;
for(int i=1 ; i<=n ; i++)
{
cout<<f3[i]<<" ";
}
cout<<endl;*/
for(int i=1 ; i<=n ; i++)
{
f1[i]=f1[i-1]+f1[i];
f2[i]=f2[i-1]+f2[i];
f3[i]=f3[i-1]+f3[i];
}
int max1=0;
for(int i=1 ; i<=n-k+1 ; i++)
{
int ans1=f1[i+k-1]-f1[i-1];
int ans2=f2[i+k-1]-f2[i-1];
int ans3=f3[i+k-1]-f3[i-1];
max1=max(max(max(ans1,ans2),ans3),max1);
}
cout<<k-max1<<endl;
}
return 0;
}
总结
这一题是我看了别人的题解才补出来的,因为区间是固定的,所以只需要求出每一个固定区间中的贡献就可以,而求区间的和肯定用前缀和来维护,想到固定区间的时候,以后一定要想到怎么可以O(1)时间求出区间的值。