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