lightoj 1363 - Binary Matrix (II) 贪心

给定一个01矩阵，每次可以交换相邻两列的一个点或者相邻两行的一个点(第一行和最后一行也算相邻，同理列也是)，问能否使矩阵变为以下三种情况之一:1.每行每列的1都相等2.每行的1都相等.3.每列的1都相等，或者不可能，对于前三种情况输出最小步数。

判断的话先统计出来所有的1，如果1的个数是行数的倍数就是行可以，是列数的倍数就是列可以，同时成立就是第一种..

那么判断完成立后我们需要找出来最小步数，因为是个环...我开始只考虑0行与n-1行到底要不要交换错了。

考虑到环的情况，以前做可以循环移动的环形都是复制一遍加后面，那么我也复制了一遍加到后面，枚举起点找出最小..

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define ull unsigned long long
#define mod 1000007
#define inf 0x3f3f3f3f
#define N 100100
using namespace std;
char mp[1234];
int r[1234],c[1234],n,m,b[2234];
int js(int q,int t,int x)
{
    int ans=0;
    for(int i=0;i<t;i++)
    {
        ans+=abs(x-b[i+q]);
        b[i+1+q]-=x-b[i+q];
    }
    return ans;
}
int pd(int a[],int t,int x)
{
    int ans=inf;
    for(int qd=0; qd<t; qd++)//枚举起点
    {
        for(int i=0; i<t; i++)
            b[i]=b[i+t]=a[i];
        ans=min(ans,js(qd,t,x));
    }
    return ans;
}
int main()
{
    int t;
    scanf("%d",&t);
    for(int cas=1; cas<=t; cas++)
    {
        int sum=0;
        scanf("%d %d",&n,&m);
        memset(r,0,sizeof(r));
        memset(c,0,sizeof(c));
        for(int i=0; i<n; i++)
        {
            scanf("%s",mp);
            for(int j=0; j<m; j++)
            {
                if(mp[j]=='1')
                {
                    r[i]++;
                    c[j]++;
                    sum++;
                }
            }
        }
        int ans=0,k=0;
        if(sum%n==0)
        {
            ans+=pd(r,n,sum/n);
            k++;
        }
        if(sum%m==0)
        {
            ans+=pd(c,m,sum/m);
            k+=2;
        }
        printf("Case %d: ",cas);
        if(k==0)
            printf("impossible\n");
        else if(k==1)
            printf("row %d\n",ans);
        else if(k==2)
            printf("column %d\n",ans);
        else
            printf("both %d\n",ans);
    }
    return 0;
}