POJ3276-Face The Right Way

这是一道反转开关问题，想让所有的牛都面朝前方，令F为0， B为1，给定一个k值，可以从最左端的牛开始，如果这头牛面朝后方，那么以这头牛为开始往后的k头牛的区间都应该反转，再以下一头牛为起点，以此类推…
用f[i]记录以i为起点的牛是否反转，反转为1，否则为0.
同时用sum记录第i头牛的反转情况，如果sum为奇数则与原方向相反，用sum与原方向相加即可得出是否需要反转。
当所有区间都反转完毕后需要检验剩下的k-1头牛是否面向前方。

#include 
#include 

const int maxn = 5000 + 10;

int dir[maxn];
int f[maxn];

int calc(int K, int n) {
    memset(f, 0, sizeof(f));
    int res = 0;
    int sum = 0;
    for (int i = 0; i + K <= n; i++) {
        if ((dir[i] + sum) % 2 != 0) {
            res++;
            f[i] =  1;
        }
        sum += f[i];
        if (i - K + 1 >= 0) {
            sum -= f[i-K+1];
        }
    }

    for (int i = n - K + 1; i < n; i++) {
        if ((dir[i] + sum) % 2 != 0) {
            return -1;
        }
        if (i - K + 1 >= 0) {
            sum -= f[i-K+1];
        }
    }

    return res;
}

int main(int argc, char const *argv[]) {
    int n;
    scanf("%d", &n);
    getchar();
    for (int i = 0; i < n; i++) {
        char ch;
        scanf("%c", &ch);
        getchar();
        if (ch == 'F') {
            dir[i] = 0;
        } else {
            dir[i] = 1;
        }
    }
    int K = 1, M = n;
    for (int k = 1; k <= n; k++) {
        int m = calc(k, n);
        if (m >= 0 && M > m) {
            M = m;
            K = k;
        }
    }
    printf("%d %d\n", K, M);

    return 0;
}