模拟,每次从左到右找第一个最优插入点,更新新的字符串,当本字符串已经能够得到最优值6,那么之后不用在模拟了,直接计算输出即可。
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
#include<vector>
using namespace std;
char a[55];
int judge(char& a){ //判断哪种颜色
if(a=='R') return 0;
else if(a=='B') return 1;
else if(a=='G') return 2;
}
int main(){
int T;
scanf("%d",&T);
getchar();
while(T--){
scanf("%s",a);
int n = strlen(a);
char t[55];
int l = 0;
int ans = 0;
for(int i=0;i<n;i++){
int d = 0; //最后插入的位置
int sum = 0; //本次最优的值
int left = 0; //左边的值
int right = 0; //右边的值
for(int j = 0;j<l;j++){ //对每一个位置枚举
left = right = 0; //每次初始化
char s[3]; //每个位计数
memset(s,0,sizeof(s));
for(int x = 0;x<j;x++){ //计算左边的颜色种数
if(s[judge(t[x])]==0){
s[judge(t[x])]=1;
left++;
}
if(left==3) break; //当已经为三种,最大值了,则不需要再判断
}
memset(s,0,sizeof(s));
for(int x = j;x<l;x++){ //计算右边的颜色种数
if(s[judge(t[x])]==0){
s[judge(t[x])]=1;
right++;
}
if(right==3) break;
}
if(sum<left+right){ //若当前点插入最优,则更换
sum = left+right;
d = j;
}
}
if(sum == 6){ //当本次左右的值都为最大值,则之后在本位置插入都会为最大值,无需在模拟
ans += (n-i)*6;
break;
}
else{
ans += sum;
}
for(int k=l;k>=d&&k-1>=0;k--){ //更新当前序列
t[k] = t[k-1];
}
t[d] = a[i];
l++;
}
printf("%d\n",ans);
}
return 0;
}