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;
}