反转 开关问题

#include<iostream>
#include<memory.h>
using namespace std;
int N,dir[5005];
int j[5005];   //标记区间[i,i-K+1]是否反转 反转则为1, 不反转为0
int calc(int K)
{
    memset(j,0,sizeof(j));
    int res=0; //反转次数
    int sum=0; //j的和 用来表示当前块之前所经历的反转次数
    for(int i=0;i+K<=N;i++)
    {
        //计算区间[i,i-K+1]
        if((dir[i]+sum)%2!=0)//需要进行反转
        {
            j[i]=1;
            res++;
        }
        sum+=j[i];
        if(i-K+1>=0)
        {
            //保持区间长度
            sum-=j[i-K+1];
        }
    }

    //检查剩下的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-=j[i-K+1];
        }
    }
    return res;
}
int main()
{
    cin>>N;
    char c;
    for(int i=0;i<N;i++)
    {
        //牛的方向 0:F 1:B
        cin>>c;
        if(c==‘F‘) dir[i]=0;
        else dir[i]=1;
    }
    //最少操作次数 与之对应的最小操作长度
    int M=N,K=1;
    for(int i=1;i<=N;i++)
    {
        int x=calc(i);
        if(x>0&&x<M)
        {
            K=i;
            M=x;
        }
    }
    cout<<K<<" "<<M<<endl;
    return 0;
}

#include<iostream>
#include<memory.h>
using namespace std;
const int dx[5]= {-1,0,0,0,1};
const int dy[5]= {0,-1,0,1,0};
int M,N;
int tile[16][16]; //起始
int opt[16][16];  //保存最优解
int flip[16][16]; //保存中间结果

//查询(x,y)的颜色
int get(int x,int y)
{
    int c=tile[x][y];
    for(int i=0; i<5; i++)
    {
        int x2=x+dx[i];
        int y2=y+dy[i];
        if(x2>=0&&y2>=0&&x2<M&&y2<N)
        {
            c+=flip[x2][y2];
        }
    }
    return c%2;
}

//求出第一行确定情况下的最小操作次数
//不存在解的话返回-1
int calc()
{
    //求出从第二行开始的翻转方法
    for(int i=1; i<M; i++)
    {
        for(int j=0; j<N; j++)
        {
            if(get(i-1,j)!=0)
            {
                //(i-1,j)是黑色的话 必须翻转这个格子
                flip[i][j]=1;
            }
        }
    }

    //判断最后一行是否全白
    for(int j=0; j<N; j++)
    {
        if(get(M-1,j)!=0)
        {
            return -1; //无解
        }
    }

    //统计翻转的次数
    int res=0;
    for(int i=0; i<M; i++)
    {
        for(int j=0; j<N; j++)
        {
            res+=flip[i][j];
        }
    }
    return res;
}

int main()
{
    cin>>M>>N;
    for(int i=0; i<M; i++)
    {
        for(int j=0; j<N; j++)
        {
            cin>>tile[i][j];
        }
    }

    int res=-1;
    //按照字典序尝试第一行的所有可能性
    for(int i=0; i<1<<N; i++)
    {
        memset(flip,0,sizeof(flip));
        for(int j=0; j<N; j++)
        {
            flip[0][N-j-1]=i>>j&1;
        }
        int num=calc();
        if(num>=0&&(res<0||res>num))
        {
            res=num;
            memcpy(opt,flip,sizeof(flip));
        }
    }

    if(res<0)
    {
        //无解
        cout<<"IMPOSSIBLE"<<endl;
    }
    else
    {
        for(int i=0; i<M; i++)
        {
            for(int j=0; j<N; j++)
            {
                cout<<opt[i][j]<<" ";
            }
            cout<<endl;
        }
    }
    return 0;
}