AT4514 [AGC030E] Less than 3

https://www.luogu.com.cn/problem/AT4514

往01中间插一根红线，10中间插一根蓝线，如上图
然后匹配，计算距离即可，可以证明一定可以取到下界

具体实现枚举偏移量，然后大力计算

code:

#include<bits/stdc++.h>
#define N 200050
using namespace std;
int n, gs, gss, a[N], b[N];
char st[N], stt[N];
int main() {
    scanf("%d", &n);
    scanf("%s", st + 1);
    scanf("%s", stt + 1);
    for(int i = 1; i < n; i ++) if(st[i] != st[i + 1]) a[++ gs] = i;
    for(int i = 1; i < n; i ++) if(stt[i] != stt[i + 1]) b[++ gss] = i;

    int ans = 1e9;
    for(int py = - gs - 1; py <= gss + 1; py ++) if((py & 1) ^ (st[1] == stt[1])) {
        int ret = 0;
        for(int i = min(1, 1 - py); i <= max(gs, gss - py); i ++) {
            int j = i + py;
            ret += abs(
                (i < 0 ? 0 : (i <= gs ? a[i] : n))
                -
                (j < 0 ? 0 : (j <= gss ? b[j] : n))
            );
        }
        ans = min(ans, ret);
    }
    printf("%d", ans);
    return 0;
}