[POJ 3276] Face The Right Way （翻转问题+技巧）

POJ - 3276
有 N个奶牛排成一列，有些朝前有些朝后，
一次可以翻转连续的 K头奶牛的朝向，K是固定的
求最少翻转次数，以及最小的 K

这种翻转的问题有两个很明显的性质
1) 翻转的位置先后是无关的
2) 同一个地方最多翻转一次

于是枚举 K，再枚举 N依次去翻转
K长度的窗口依次向右滑，如果此时最左边的牛朝后
则一定要翻转，因为此后的操作都与其无关了

不过直接搞的话时间复杂度是 O(n^3)
用到一点小技巧，就是用 rev标记此时是否要翻转
如果翻转，则 rev^=1，并用 note数组标记此次翻转退出的地点
这样时间就优化到了 O(n^2)

有个坑点：
由于我 i从 K开始，一直到 N
所以如果走到 N的时候，还需额外的一个循环判断 N-K+2..N 的是否能翻转完
不仅如此，剩下的翻转退出标记也要一一算到

#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <map>
#include <set>
#include <queue>
using namespace std;
typedef pair<int,int> Pii;
typedef long long LL;
typedef unsigned long long ULL;
typedef double DBL;
typedef long double LDBL;
#define MST(a,b) memset(a,b,sizeof(a))
#define CLR(a) MST(a,0)
#define Pow2(a) (a*a)

const int maxn=5e3+10;
int N;
bool inpt[maxn];
bool note[maxn];

int main()
{
    while(~scanf("%d", &N))
    {
        CLR(inpt);
        for(int i=1; i<=N; i++)
        {
            char chr;
            scanf(" %c", &chr);
            if(chr=='B') inpt[i]=1;
        }
        int ansk=1,ans=N+10;
        for(int K=1; K<=N; K++)
        {
            int cnt=0;
            bool rev=0;
            CLR(note);
            for(int i=K; i<=N; i++)
            {
                bool now=inpt[i-K+1]^rev;
                if(now)
                {
                    note[i]=1;
                    cnt++;
                    rev^=1;
                }
                rev^=note[i-K+1];
            }
            bool ok=1;
            for(int i=N-K+2; i<=N; i++)
            {
                if(inpt[i]^rev){ok=0;break;}
                rev^=note[i];
            }
            if(ok&&cnt<ans)
            {
                ansk=K;
                ans=cnt;
            }
        }
        printf("%d %d\n", ansk, ans);
    }
    return 0;
}