[AGC030E]Less than 3

Description

給出兩個長度爲n的01串s和t
你進行若干次操作，每次操作可以更改s中某一個位置上的值，每次操作前後需要保證s中不存在相鄰3個一樣的字符
現在問把s變爲t所需要的最小操作次數
n<=5000

Solution

niubi題
我們在0->1之間畫一條紅線，在1->0之間畫一條藍線
默認字符串開頭結尾有無限多的紅藍線，我們可以發現兩個字符串相等等價於這兩個字符串的紅藍線相等
然後會發現更改一個位置上的數相當於將一條紅藍線左移/右移
並且時時刻刻相鄰紅藍線之間的距離<=2
那麼我們可以直接枚舉紅藍線之間的對應關係暴力計算答案

Code

#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;

const int N=2e4+5;

int n,a[N],b[N];
char s[N],t[N];

int calc() {
	int t1=0,t2=0;
	fo(i,1,n-1) if (s[i]=='0'&&s[i+1]=='1') a[++t1]=i;
	fo(i,1,n) b[++t2]=0;
	fo(i,1,n-1) if (t[i]=='0'&&t[i+1]=='1') b[++t2]=i;
	fo(i,1,n) b[++t2]=n;
	int ret=n*n;
	fo(i,1,t2-t1+1) {
		int now=0;
		fo(j,1,i-1) now+=b[j];
		fo(j,1,t1) now+=abs(a[j]-b[i+j-1]);
		fo(j,i+t1,t2) now+=n-b[j];
		ret=min(ret,now);
	}
	return ret;
}

int main() {
	scanf("%d",&n);scanf("%s",s+1);scanf("%s",t+1);
	if (n<=2) {
		int ans=0;
		fo(i,1,n) ans+=s[i]!=t[i];
		printf("%d\n",ans);
		return 0;
	}
	int tmp=calc();
	fo(i,1,n) s[i]='0'+'1'-s[i],t[i]='0'+'1'-t[i];
	printf("%d\n",tmp+calc());
	return 0;
}