AtCoder agc 030 E

Less than 3

题目链接

Solution

首先我们可以发现一个性质，假设我们要变动第$i$个位置上的字符，可以发现如果变动使得变动前后不存在连着相同的三个字符，那么一定有$i-1$位置上的字符和$i+1$位置上的字符不同。
我们在$0$和$1$之间划一条蓝分割线，$1$和$0$之间划一条红分割线，在变动过程中，我们可以发现，以下两个性质：$1$、每次变动相当于左/右移一条红/蓝分割线。$2$、任意时刻必须满足任意两条分割线之间的距离不能超过$2$。
同时我们还可以发现，一条分割线可能会移出界。
到了这里，正解思路便很清晰了，枚举$s/t$的$0$位置前面有多少条分割线，之后便可以一一匹配分割线算代价，代价最小的便是答案。

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
 
#define fo(i,j,l) for(int i=j;i<=l;++i)
#define fd(i,j,l) for(int i=j;i>=l;--i)
 
using namespace std;
typedef long long ll;
const ll N=152e2;
 
int n,m;
char s[N],t[N];
 
int a[N][2],b[N][2];
 
inline int abs(int o)
{return (o<0)?(-o):o;}
 
inline int min(int a,int b)
{return a<b?a:b;}
 
int main()
{
	scanf("%d",&n);
	scanf("%s%s",s+1,t+1);
	int u=0,v=0;
	s[0]=t[0]='.';
	fo(i,1,n)if(s[i]!=s[i-1])a[++u][1]=s[i]^48,a[u][0]=i;
	fo(i,1,n)if(t[i]!=t[i-1])b[++v][1]=t[i]^48,b[v][0]=i;
	
	int ans=n*n*2;
	
	fo(i,0,n){
		int zd=max(i+u,v);
		int lj=0,ok=1;
		fo(l,1,zd){
			int po=(l<=v)?b[l][0]:n+1;
			int co=(l<=v)?b[l][1]:2;
			int _po=(l<=i)?1:((l>i+u)?n+1:a[l-i][0]);
			int _co=(l<=i)?2:((l>i+u)?2:a[l-i][1]);
			if(_co!=co&&co!=2&&_co!=2)ok=0;
			lj=lj+abs(po-_po);
		}
		if(ok)ans=min(ans,lj);
	}
	
	u=v=0;
	fo(i,1,n)swap(s[i],t[i]);
	fo(i,1,n)if(s[i]!=s[i-1])a[++u][1]=s[i]^48,a[u][0]=i;
	fo(i,1,n)if(t[i]!=t[i-1])b[++v][1]=t[i]^48,b[v][0]=i;
	
	fo(i,0,n){
		int zd=max(i+u,v);
		int lj=0,ok=1;
		fo(l,1,zd){
			int po=(l<=v)?b[l][0]:n+1;
			int co=(l<=v)?b[l][1]:2;
			int _po=(l<=i)?1:((l>i+u)?n+1:a[l-i][0]);
			int _co=(l<=i)?2:((l>i+u)?2:a[l-i][1]);
			if(_co!=co&&co!=2&&_co!=2)ok=0;
			lj=lj+abs(po-_po);
		}
		if(ok)ans=min(ans,lj);
	}
	
	cout<<ans;
}