【反转问题】POJ - 3276 Face The Right Way

Problem Description

N头牛排成了一列。每头牛或者向前或者向后。为了让所有的牛都面向前方，农夫约翰买了一台自动转向的机器。这个机器在购买时就必须设定一个数值K，机器每操作一次恰好使K头连续的牛转向。请求出为了让所有牛都能面向前方需要的最少的操作次数M和对应的最小的K。

思路：

首先交换区间反转的顺序对结果是没有影响的，此外，可以知道对同一个区间进行两次以上的反转是多余的。
我们枚举K，然后从左往右，遇到需要反转的就给它反转，然后标记一下区间【i, i+k-1】被反转过。核心点就是在于如何标记能实现O(1)的复杂度。
用一个f[i]数组，f[i]: = 区间[i, i+k-1]进行了反转的话则为1，否则为0。这样，在考虑第i头牛时，如果区间[i-k+1,i-1]f[i]的和为奇数的话，则这头牛的方向与起始方向是相反的

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int inf = 0x3f3f3f3f;
const int N = 5005;
int dir[N];//0面向前方，1面向后方
int f[N], n;
int cal(int k)
{
    memset(f, 0, sizeof(f));
    int sum = 0, res = 0;
    for(int i = 0; i + k - 1 < n; i++)
    {
        if(i-k >= 0) {//[i-k+1,i-1]的和影响，所以减去f[i-k]
            sum -= f[i-k];
        }
        if((dir[i]+sum)%2 != 0) {//sum偶数i这个位置不变 sum奇数i这个位置相反
            res++;
            f[i] = 1;
        }
        sum += f[i];
    }
    for(int i = n-k+1; i < n; i++)//检查剩下的牛是否有面朝后方
    {
        if(i-k >= 0) {
            sum -= f[i-k];
        }
        if((dir[i]+sum)%2 != 0) {
            return -1;
        }
    }
    return res;
}
void solve()
{
    int K, M = inf;
    for(int k = 1; k <= n; k++)//枚举一次可以反转的区间，题目要求从1开始。
    {
        int m = cal(k);
        if(m != -1 && M > m)
        {
            K = k, M = m;
        }
    }
    printf("%d %d\n", K, M);
}
int main()
{
    char s[2];
    while(~scanf("%d", &n))
    {
        for(int i = 0; i < n; i++) {
            scanf("%s", s);
            if(s[0] == 'F') dir[i] = 0;
            else dir[i] = 1;
        }
        solve();
    }
    return 0;
}