CodeForces - 1196D2 - RGB Substring (hard version) （ 思维 + 前缀和 ）

题目链接：点击进入

题目

题意

思路

先来暴力想一下，长度为k的串且是无限长串的子串，只有三种 ( 分别以 R 、G 、B 开头的三种 )，暴力求解就是以 s 的每个位置为子串起点，暴力匹配三种子串，时间复杂度 3 * n * k 。这肯定是不行的，中间匹配 k 的过程太大，需要优化。
那我们来想一下，对于一种子串 ( 假设以 R 开头 ) 将它延长到 | s | ，然后将它与 s 匹配，记录两者不相同的位置。对于前 k 个，前缀和求出来的就是至少需要修改的数量，但是大于 k 的位置了咋办。若是大于 k 的位置 ( j = k + 1 ) 我们还用前缀和求 ( sum [ j ]- sum [ j - k ] )，那么求出来的就不是以 R 开头的了，这时候求出来的就是以 G 开头的了。那么我们咋办。
其实大家想一下，这里求出来是以 G 开头的，那么对于另一种以 G 开头的子串求出来的就是以 B 开头的，以 B 开头的子串求出来的就是以 R 开头的，三者相互交错了，但是没有遗漏。也就是说，我们对于三种子串都扩充到 | s | ，不同位置 +1 ，求前缀和，做长度为k的前缀和差值，就可以求得所有位置为起点，与分别以 R 、G 、 B 开头的长度为 k 的子串的匹配情况。答案取 min 即可。

代码

// Problem: RGB Substring (hard version)
// Contest: Virtual Judge - CodeForces
// URL: https://vjudge.net/problem/CodeForces-1196D2
// Memory Limit: 262 MB
// Time Limit: 2000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

//#pragma GCC optimize(3)//O3
//#pragma GCC optimize(2)//O2
#include<iostream>
#include<string>
#include<map>
#include<set>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<stack>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<fstream>
#define X first
#define Y second
#define best 131 
#define INF 0x3f3f3f3f3f3f3f3f
#define pii pair<int,int>
#define lowbit(x) x & -x
#define inf 0x3f3f3f3f
//#define int long long
//#define double long double
//#define rep(i,x,y) for(register int i = x; i <= y;++i)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double pai=acos(-1.0);
const int maxn=1e6+10;
const int mod=1e9+7;
const double eps=1e-9;
const int N=5e3+10;
/*--------------------------------------------*/
inline int read()
{
    int k = 0, f = 1 ;
    char c = getchar() ;
    while(!isdigit(c)){if(c == '-') f = -1 ;c = getchar() ;}
    while(isdigit(c)) k = (k << 1) + (k << 3) + c - 48 ,c = getchar() ;
    return k * f ;
}
/*--------------------------------------------*/

int t,n,k,sum[maxn];
char s[maxn],str[3][4]={"RGB","GBR","BRG"};

int main() 
{
//	ios::sync_with_stdio(false);
//	cin.tie(0);cout.tie(0);
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d",&n,&k);
		scanf("%s",s+1);
		int ans=inf;
		for(int i=0;i<3;i++)
		{
			for(int j=1;j<=n;j++) sum[j]=0;
			for(int j=1;j<=n;j++)
				if(s[j]!=str[i][j%3]) 
					sum[j]++;
			for(int j=1;j<=n;j++)
			{
				sum[j]=sum[j]+sum[j-1];
				if(j>=k) ans=min(ans,sum[j]-sum[j-k]);
			}
		}
		printf("%d\n",ans);
	}
    return 0;
}