UVALive 5809 Binary Matrix

题意:给出一个只有0和1的矩阵，相邻的2个位置（包括头和尾）可以交换，问通过最少的交换次数可以让每一行和每一列的1数目相同

思路:刚开始觉的要同时照顾行和列进行转换会很麻烦，看了别人的题解和代码，其实只要分别对行和列转换使其分别满足条件就可以了，要怎么使交换步数最少呢，参照某一道经典的换苹果题目，将其转换为到某一点距离最短，由于数据小，可以暴力一遍（不暴力用某个神奇的公式似乎也可以……但是没看懂），也有人用费用流做，同样不懂+_+

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
int row[1030],col[1030],tmp[1030],n,m;
char mp[1030][1030];
int change(int now,int gold,int n)
{
    int N=n,ans=0;
    while (N--)
    {
        if (tmp[now]>gold)
        {
            ans+=tmp[now]-gold;
            tmp[(now+1)%n]+=tmp[now]-gold;
        }
        else if (tmp[now]<gold)
        {
            ans+=gold-tmp[now];
            tmp[(now+1)%n]-=gold-tmp[now];
        }
        now=(now+1)%n;
    }
    return ans;
}

int cal(int s)
{
    int res=0,flag1=0,flag2=0;;
    if (s%n==0)
    {
        flag1=1;
        int ans=0x3f3f3f3f;
        for (int i=0;i<n;i++)
        {
            for (int i=0;i<n;i++)
             tmp[i]=row[i];
            ans=min(ans,change(i,s/n,n));
        }
        res+=ans;
    }
    if (s%m==0)
    {
        flag2=1;
        int ans=0x3f3f3f3f;

        for (int i=0;i<m;i++)
        {
            for (int i=0;i<m;i++)
             tmp[i]=col[i];
            ans=min(ans,change(i,s/m,m));
        }
        res+=ans;
    }
   if (flag1==1 && flag2==1) printf("both %d\n",res);
   else if (flag1==1) printf("row %d\n",res);
   else if (flag2==1) printf("column %d\n",res);
   else printf("impossible\n");
}

int main()
{
    int t,cas=0;
    scanf("%d",&t);
    while (t--)
    {
        cas++;
        int sum=0;
        memset(row,0,sizeof(row));
        memset(col,0,sizeof(col));
        scanf("%d%d",&n,&m);
        for (int i=0;i<n;i++)
         scanf("%s",&mp[i]);

        for (int i=0;i<n;i++)
         for (int j=0;j<m;j++)
         {
             if (mp[i][j]=='1')
             {
                 row[i]++;
                 col[j]++;
                 sum++;
             }
        }
        printf("Case %d: ",cas);
        cal(sum);
    }
}