AtCoder Beginner Contest 162 D RGB Triplets 前缀和

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;
}