AtCoder Beginner Contest 162 比赛人数10673 快,比赛开始后5分钟看到所有题
AtCoder Beginner Contest 162 D RGB Triplets 前缀和
总目录详见https://blog.csdn.net/mrcrack/article/details/104454762
在线测评地址https://atcoder.jp/contests/abc162/tasks/abc162_d
因1≤N≤4000,采用O(n^2)的做法。
要寻找的字母,存在6种组合:RGB,BGR,RBG,GBR,BRG,GRB
j−i≠k−j.注意2*j!=i+k,此句理解,相见手工算法
手工算法模拟如下
4
RRGB
1
位置 1 2 3 4
字符串 R R G B
1 4
R B
R与B之间,字母G的数量,可以用前缀和维护
(1+4)/2不是整数,故
1 3 4
R G B
符合题意
2 4
R B
(2+4)/2=3,
故
2 3 4
R G B
这组数据不符合题意。
比赛时,写的手忙脚乱,6组数据,竟然写了6次雷同的代码,以下代码,为去除冗余后的代码,为了增强代码可读性,还有冗余的可去除,就保留了。
#include <stdio.h>
#include <string.h>
#define maxn 4010
#define LL long long
LL sum;
char s[maxn];
int r[maxn],g[maxn],b[maxn],n;
void count(char a,char b,char c,int *p){//abc
int i,j,k;
for(i=1;i<=n;i++)
for(k=i+1;k<=n;k++)
if(s[i]==a&&s[k]==c){
if((i+k)%2==0){
j=(i+k)/2;
if(s[j]==b)sum+=p[k-1]-p[i]-1;
else sum+=p[k-1]-p[i];
}else sum+=p[k-1]-p[i];
}
}
int main(){
int i;
scanf("%d%s",&n,s+1);
for(i=1;i<=n;i++){
r[i]=r[i-1],g[i]=g[i-1],b[i]=b[i-1];
if(s[i]=='R')r[i]+=1;
else if(s[i]=='G')g[i]+=1;
else if(s[i]=='B')b[i]+=1;
}
count('R','G','B',g);
count('B','G','R',g);
count('R','B','G',b);
count('G','B','R',b);
count('B','R','G',r);
count('G','R','B',r);
printf("%lld\n",sum);
return 0;
}