Codeforces1913 E. Matrix Problem（最小费用最大流，思维好题）

题意：

解法：

sum(A[])!=sum(B[])时无解.

考虑最小费用最大流(数据范围小+求最小代价+网格图).

存在两个约束:A[]和B[],并且需要满足sum(A[])=sum(B[]).
令R[i]为第i行的行节点,
令C[j]为第j劣的列节点,
左半部建图:令S->R[i],容量为A[i],费用为0,
右半部建图:令C[j]->T,溶解为B[j],费用为0.
这样的话只要满流就能满足题目条件.

中部建图:一个错误的想法是:
对于a[i][j]=1的点,R[i]->C[j],容量为1,费用为0,
对于a[i][j]=0的点,R[i]->C[j],容量为1,费用为1.
为什么说这样的做法是错误的呢?
因为这样建图只考虑了0->1的代价,没有考虑1->0的代价.
(也可以理解为在没有流量时,没有将1边置为0边.只满足了>=A[],>=B[]条件,不满足=的条件)

一个很巧妙的办法是:
我们首先将网格图中所有1的格子翻转为0,记录代价为cnt1.
对于a[i][j]=0的边,令R[i]->C[j],容量为1,费用为1,
对于a[i][j]=0的边,令R[i]->C[j],容量为1,费用为-1.
这样的话只有如果流过了费用为-1的边，相对于还原了一个原来的1位置,
如果流过了费用为1的边,相对于翻转了一个0位置.
最后答案为cnt1+mincost

Code：

#include <bits/stdc++.h>
using namespace std;
#define X first
#define Y second
#define int long long
#define PI pair<int, int>
const int maxm=3e5+5;
const int mod=998244353;

const int N=1e5+5;
const int M=6e5+5;
const int inf=1e9;
struct Node{
    int from,to,nt,cap,flow,cost;
}e[M];
int head[N],cnt=1;
int pre[N],mark[N],d[N];
int S,T;
int mincost,maxflow;
int idx=0;
void init(){
  for(int i=0;i<=idx;i++)head[i]=0;
  cnt=1;
  idx=0;
}
void add(int a,int b,int c,int d){
    cnt++;e[cnt].nt=head[a];head[a]=cnt;e[cnt].from=a;e[cnt].to=b;
    e[cnt].cap=c;e[cnt].flow=0;e[cnt].cost=d;
    cnt++;e[cnt].nt=head[b];head[b]=cnt;e[cnt].from=b;
    e[cnt].to=a;e[cnt].cap=0;e[cnt].flow=0;e[cnt].cost=-d;
}
bool spfa(){
  for(int i=0;i<=idx;i++)mark[i]=0;
  for(int i=0;i<=idx;i++)d[i]=inf;
  for(int i=0;i<=idx;i++)pre[i]=0;
  queue<int>q;
  q.push(S);
  d[S]=0;
  mark[S]=1;
  while(!q.empty()){
      int x=q.front();
      q.pop();
      mark[x]=0;
      for(int i=head[x];i;i=e[i].nt){
          int v=e[i].to;
          if(e[i].cap>e[i].flow&&d[v]>d[x]+e[i].cost){
              d[v]=d[x]+e[i].cost;
              pre[v]=i;
              if(!mark[v]){
                  q.push(v);
                  mark[v]=1;
              }
          }
      }
  }
  return d[T]!=inf;
}
void ek(){
  mincost=maxflow=0;
  while(spfa()){
    int k=inf;
    for(int i=T;i!=S;i=e[pre[i]].from){
      k=min(k,e[pre[i]].cap-e[pre[i]].flow);
    }
    for(int i=T;i!=S;i=e[pre[i]].from){
      e[pre[i]].flow+=k;
      e[pre[i]^1].flow-=k;
    }
    maxflow+=k;
    mincost+=k*d[T];
  }
}
int n,m;
int a[55][55];
int A[55],B[55];
int R[55],C[55];
void solve(){
  init();
  cin>>n>>m;
  for(int i=1;i<=n;i++){
    for(int j=1;j<=m;j++){
      cin>>a[i][j];
    }
  }
  for(int i=1;i<=n;i++)cin>>A[i];
  for(int i=1;i<=m;i++)cin>>B[i];
  int sa=0;
  int sb=0;
  for(int i=1;i<=n;i++)sa+=A[i];
  for(int j=1;j<=m;j++)sb+=B[j]; 
  if(sa!=sb){
    cout<<-1<<endl;
    return ;
  }
  // build graph.
  S=++idx;
  T=++idx;
  for(int i=1;i<=n;i++)R[i]=++idx;
  for(int j=1;j<=m;j++)C[j]=++idx;
  for(int i=1;i<=n;i++){
    add(S,R[i],A[i],0);
  }
  for(int j=1;j<=m;j++){
    add(C[j],T,B[j],0);
  }
  int ans=0;
  for(int i=1;i<=n;i++){
    for(int j=1;j<=m;j++){
      if(a[i][j]){
        add(R[i],C[j],1,-1);
        ans++;
      }else{
        add(R[i],C[j],1,1);
      }
    }
  }
  ek();
  if (maxflow!=sa){
    cout<<-1<<endl;
    return ;
  }
  ans+=mincost;
  cout<<ans<<endl;
}
signed main() {
// #define MULTI_CASE
  ios::sync_with_stdio(0);
  cin.tie(0);
#ifndef ONLINE_JUDGE
  freopen("../in.txt", "r", stdin);
  freopen("../out.txt", "w", stdout);
#endif
#ifdef MULTI_CASE
  int T;
  cin >> T;
  while (T--)
#endif
    solve();
  return 0;
}

---