标签：
第一行一个整数K，1<=K<=3000.
第二行一个字符串S，1<=len（S）<=3000,且只由‘R‘,‘G‘,‘B‘三种字符组成。
一个整数，即不同的四元组(a,b,c,d)的个数。
容斥+dp，
ans=([a,b]存在k个连续的‘G‘，[c,d]任意的方案数)+([c,d]存在k个连续的‘G‘，[a,b]任意的方案数)-([a,b],[c,d]都存在k个连续的‘G‘的方案数)+([a,b],[c,d]都不存在k个连续的‘G‘，但拼起来之后存在k个连续的‘G‘的方案数)
#include<cstdio> #include<cstring> char s[3003]; int n,k; long long ans=0; int ls[3003],rs[3003]; short cl[3003][3003],cr[3003][3003],cs[3003][3003]; int l1[3003][3003],r1[3003][3003]; int max(int a,int b){return a>b?a:b;} int C(int x){return x*(x+1)/2;} int main(){ scanf("%d%s",&k,s+1); n=strlen(s+1); for(int i=1;i<=n;++i)if(s[i]!=‘G‘)s[i]=0; for(int i=1;i<=n;++i)if(s[i])cs[i][i]=cl[i][i]=cr[i][i]=1; for(int d=1;d<n;++d)for(int l=1,r=1+d;r<=n;++l,++r){ if(s[r])cl[l][r]=cl[l][r-1]+1; if(s[l])cr[l][r]=cr[l+1][r]+1; cs[l][r]=max(max(cs[l][r-1],cs[l+1][r]),max(cl[l][r],cr[l][r])); } for(int i=1;i<=n;++i){ for(int j=1;j<=i;++j)if(cs[j][i]>=k)++ls[i]; for(int j=i;j<=n;++j)if(cs[i][j]>=k)++rs[i]; } for(int i=n;i;--i)rs[i]+=rs[i+1]; for(int i=1;i<n;++i)ans+=1ll*ls[i]*C(n-i)+1ll*rs[i+1]*i-1ll*ls[i]*rs[i+1]; for(int i=1;i<=n;++i)for(int j=i;j<=n;++j){ if(cs[i][j]<k&&cs[i][j]){ if(cl[i][j])++l1[j][cl[i][j]]; if(cr[i][j])++r1[i][cr[i][j]]; } } for(int i=n;i;--i)for(int j=1;j<k;++j)r1[i][j]+=r1[i+1][j]; for(int i=1;i<=n;++i)for(int j=k-1;j;--j)r1[i][j]+=r1[i][j+1]; for(int i=1;i<n;++i)for(int j=1;j<k;++j)ans+=1ll*l1[i][j]*r1[i+1][k-j]; printf("%lld",ans); return 0; }
第一行一个整数K，1<=K<=3000. 第二行一个字符串S，1<=len（S）<=3000,且只由‘R‘,‘G‘,‘B‘三种字符组成。
一个整数，即不同的四元组(a,b,c,d)的个数。
标签：
原文地址：http://www.cnblogs.com/ccz181078/p/5858152.html
