bzoj1458 士兵占领

　　费用流,连下面几类边

1.s->s'，流量为n*m，费用为0，表示最多可放置n*m个士兵

2.s'->行

(1)流量为a[i]，费用为-n*m，表示必须在这一行放置a[i]个士兵。

(2)流量为n*m,费用为0，表示该行可以放置其他任意数目士兵。

3.行->列，流量为1，费用为1，表示第i行第j列放置的士兵数目

4.列->t

(1)流量为b[i]，费用为-n*m，表示必须在这一列放置b[i]个士兵。

(2)流量为n*m,费用为0，表示该列可以放置其他任意数目士兵。

5.s'->t，流量为n*m，费用为0，表示一些士兵可以不用放置。

最后答案为费用+(Σa[i]+Σb[i])*n*m

代码

 #include<cstdio>
 #include<algorithm>
 #include<queue>
 #define mp make_pair
 #define pii pair<int,int>
 #define N 10010
 #define M 200010
 #define inf 0x37373737
 using namespace std;
 struct MCMF{
     int h[N] , dis[N] , ing[N] , pre[N] , s , t , n;
     int to[M] , ne[M] , cap[M] , cost[M] , e;
     void ini(){
         fill(h,h+N,-);
         e = ;
     }
     void liu(int u,int v,int c,int w){
         to[e] = v , ne[e] = h[u] , cap[e] = c , cost[e] = w;
         h[u] = e++;
     }
     void link(int u,int v,int c,int w){
         liu(u,v,c,w);
         liu(v,u,,-w);
     }
     bool spfa(){
         queue<int> Q;
         fill(ing,ing+n,);
         fill(pre,pre+n,-);
         fill(dis,dis+n,inf);
         ing[s] = true , dis[s] = ;
         Q.push(s);
         while(!Q.empty()){
             int c = Q.front();Q.pop();ing[c] = false;
             for(int k=h[c];~k;k=ne[k]){
                 int v = to[k];
                 if(cap[k] <= ) continue;
                 if(dis[c] +  cost[k] < dis[v]){
                     dis[v] = dis[c] + cost[k];
                     pre[v] = k;
                     if(!ing[v]) Q.push(v) , ing[v] = true;
                 }
             }
         }
         return dis[t] != inf;
     }
     int flow , mincost;
     pii run(int _s,int _t,int _n){
         s = _s , t = _t , n = _n;
         flow = mincost = ;
         while(spfa()){
             int pl = inf , p , k;
             for(p=t;p!=s;p=to[k^]){
                 k = pre[p];
                 pl = min(pl,cap[k]);
             }
             for(p=t;p!=s;p=to[k^]){
                 k = pre[p];
                 cap[k] -= pl;
                 cap[k^] += pl;
             }
             mincost += pl * dis[t];
             flow += pl;
         }
         return mp(flow,mincost);
     }
 };
 MCMF t;
 int n,m,k,i,j,a[N],b[N],sum;
 int flag[][];
 int main()
 {
     scanf("%d%d%d",&n,&m,&k);
     for (i=;i<=n;i++)
     scanf("%d",&a[i]),sum+=a[i];
     for (i=;i<=m;i++)
     scanf("%d",&b[i]),sum+=b[i];
     for (i=;i<=k;i++)
     {
         int A,B;
         scanf("%d%d",&A,&B);
         flag[A][B]=;
     }
 
         t.ini();
         t.link(,n+m+,n*m,);
         t.link(n+m+,n+m+,n*m,);
         for (i=;i<=n;i++)
         {
             t.link(n+m+,i,a[i],-n*m);
             t.link(n+m+,i,n*m,);
         }
         for (i=;i<=m;i++)
         {
             t.link(n+i,n+m+,b[i],-n*m);
             t.link(n+i,n+m+,n*m,);
         }
         for (i=;i<=n;i++)
         for (j=;j<=m;j++)
         if (flag[i][j]==)
         t.link(i,n+j,,);
 
         int ans=t.run(,n+m+,n+m+).second;
         printf("%d\n",ans+sum*n*m);
 }