Poj 3276 Face The Right Way

链接:

点击打开链接

题意：

有N头牛朝着正面或反面，先有一个机器能够将连续的K头牛翻转，问K取多少时操作次数最少。

题解：

对于一个固定的K值我们可以枚举每头牛判断是否需要翻转，判断一头牛是否要翻转则需要知道它之前被翻转了几次与它初始状态，假如翻转次数为奇数初始状态为偶数或翻转次数为偶数初始状态为奇数则该头牛需要翻转，判断一头牛之前翻转了几次则可以在每次枚举牛时得到，假如这头牛的前K+1头牛为翻转则这头牛之前翻转的次数减一，如果这头牛需要翻转则这头牛之后的翻转次数加一，这样就可以在O（1）时间得到翻转次数，总的时间复杂度为O（N^2）

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<cmath>
#include<stdlib.h>
#include <string.h>
#include<queue>
#include<set>
#include<map>
#include<stack>
#include<time.h>
using namespace std;
#define MAX_N 5005
#define inf 0x3f3f3f3f
#define LL long long
#define ull unsigned long long
const LL INF = 9e18;
const int mod = 1e8+7;
typedef pair<LL, LL>P;

int main()
{
    int N;
    bool cow[MAX_N];
    bool flip[MAX_N];
    while(cin >> N) {
        for(int i=1; i<=N; i++) {
            char c;
            getchar();
            c = getchar();
            if(c == 'B')
                cow[i] = false;
            else
                cow[i] = true;
        }
        int K, M;
        K = 1;
        M = inf;
        //枚举K
        for(int i=1; i<=N; i++) {
            memset(flip, false, sizeof(flip));
            int sum = 0;
            int ope = 0;
            for(int j=1; j+i-1<=N; j++) {
                //判断该头牛之前翻转了几次
                if(j-i>=1 && flip[j-i]) {
                    //之前翻转次数减一
                    sum--;
                }
                //判断是否要翻转
                if(sum%2==0 && !cow[j]) {
                    ope++;
                    sum++;//之后翻转次数加一
                    flip[j] = true;
                }
                else if(sum%2==1 && cow[j]) {
                    ope++;
                    sum++;//之后翻转你次数加一
                    flip[j] = true;
                }
            }
            //检测后面i-1头牛是否都为正面
            for(int j=N+2-i; j<=N; j++) {
                if(j-i>=1 && flip[j-i]) {
                    sum--;
                }
                if(sum%2==0 && !cow[j]) {
                    ope = inf+1;
                    break;
                }
                else if(sum%2==1 && cow[j]) {
                    ope = inf+1;
                    break;
                }
            }
            //更新答案
            if(ope < M) {
                M = ope;
                K = i;
            }
            else if(ope == M) {
                K = min(K, i);
            }
        }
        cout << K << " " << M << endl;
    }
}