agc052E 3 Letters

Description

• 传送门

Solution

• 首先由于相邻的都不一样，能不能用特殊的方法表示$S$呢？如果看作是在模3的情况下，我们可以发现相邻的要么是+1要么是-1，因此$S$等价一个数组$a$，$a_i\%3=S_i$，$|a_i-a_{i-1}|=1$，在$a_1$确定的时候唯一确定。
• 而操作可以看作将$a_i$加或减2，并且要求之后还满足上面的条件。
• 现在我们要把$S$对应的$a$变成$T$对应的$b$，首先$a_1=b_1(mod2)$，因此对于所有$i$都有$a_i=b_i(mod2)$。由于$a,b$写成函数图像的优美的性质并不会在非整点相交，考虑在交点处分成若干个独立的段，不妨设$a_i>b_i$在这一段满足，那么每一次将最大值减去2一定仍满足条件，因此可以得到$ans=\sum \frac{|a_i-b_i|}{2}$
• 但是注意到$a_1,b_1$未确定，因此固定$a_1$枚举$b_1$计算最小的$ans$.预处理$a_i-b_i$的分界点讨论正负即可做到$O(n)$。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define maxn 500105
#define ll long long
using namespace std;

int n,i,j,k,a[maxn],b[maxn],S[maxn],T[maxn];
vector<int> D[maxn];
char ch;
ll ans,sum,cnt;

int main(){
	freopen("ceshi.in","r",stdin);
	scanf("%d",&n),ch=getchar();
	while (ch<'A'||ch>'C') ch=getchar();
	for(i=1;i<=n;i++) S[i]=ch-'A',ch=getchar();
	while (ch<'A'||ch>'C') ch=getchar();
	for(i=1;i<=n;i++) T[i]=ch-'A',ch=getchar();
	for(i=2;i<=n;i++) {
		if (S[i-1]+S[i]==2) a[i]=(S[i-1]>S[i])?1:-1;
		else a[i]=(S[i-1]<S[i])?1:-1;
	}
	for(i=2;i<=n;i++) {
		if (T[i-1]+T[i]==2) b[i]=(T[i-1]>T[i])?1:-1;
		else b[i]=(T[i-1]<T[i])?1:-1;
	}
	a[1]=S[1]; for(i=2;i<=n;i++) a[i]+=a[i-1];
	for(i=0;i<6;i++) if (i%3==T[1]&&((i-a[1])&1^1)) b[1]=i;
	b[1]-=(n/3+3)*6;
	for(i=2;i<=n;i++) b[i]=b[i-1]+b[i];
	for(i=1;i<=n;i++) D[(a[i]-b[i]+5)/6].push_back(i);
	for(i=1;i<=n;i++) sum+=a[i]-b[i];
	ans=sum;  cnt=n;
	for(i=1;i<=2*n/3+3;i++) {
		sum=sum-cnt*6+(n-cnt)*6;
		for(j=0;j<D[i].size();j++){
			k=D[i][j],cnt--;
			sum=sum+6-abs(a[k]-(b[k]+(i-1)*6))+abs(a[k]-(b[k]+i*6));
		}
		ans=min(ans,sum);
	}
	printf("%lld\n",ans/2);
}