POJ 3276 牛的朝向 反转问题

/*挑战程序设计》P150
N头牛，每头牛有一个状态F或者B，F面向前方，B面向后方
每次设定一个K值，代表一次可以反转K头连续的牛，问至少需要反转多少次，求出对应的K和M次
可以从1-N枚举K，求出最小的M即可
由于是一个区间[i,i+k-1]的反转，所以对于i，就有[i-k+1,i],[i-k+2,i+1]……[i,i+k-1]会影响i的
所以可以形成一种递推关系，i+1只需要在i的基础上加上[i+1,i+k],并减去[i-k+1,i];
O(n*n)即可搞定
*/

#include<cstdio>
#include<cstring>
#define FOR(i,f_start,f_end) for(int i=f_start;i<=f_end;++i)
#define MT(x,i) memset(x,i,sizeof(x))
using namespace std;
const int MAX=5005;
int N,dir[MAX],f[MAX] ;
int calc(int k)
{
    MT(f,0);
    int sum=0,res=0; //sum为i处的反转次数（也是靠它进行递推）；res为总的需要反转的次数
    for(int i=0;i+k<=N;i++){  //连续 边求f[i]边求sum
        if((dir[i]+sum)%2!=0){
            res++;//
            f[i]=1;//需要反转
        }
        sum+=f[i];                   //
        if(i-k+1>=0) sum-=f[i-k+1];  //这两个是求sum
    }

    for(int i=N-k+1;i<N;i++){ //检查剩下的牛是否都是F
        if((dir[i]+sum)%2!=0){
            return 5001; //不是则无解，返回最大值上限
        }
        if(i-k+1>=0){
            sum-=f[i-k+1];
        }
    }

    return res;//反转次数
}
void solve()
{
    int mi=N,k=1;//min，K
    for(int i=1;i<=N;i++){//枚举K
        int m=calc(i);
        if(mi>m){
            mi=m;
            k=i;
        }
    }
    printf("%d %d\n",k,mi);
}
int main()
{
    scanf("%d",&N);
    char s[3];
    for(int i=0;i<N;i++) {
        scanf("%s",s);
        if(s[0]=='B') dir[i]=1;
        else dir[i]=0;
    }
    solve();
    return 0;
}