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,-1);
        e = 0;
    }
    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,0,-w);
    }
    bool spfa(){
        queue<int> Q;
        fill(ing,ing+n,0);
        fill(pre,pre+n,-1);
        fill(dis,dis+n,inf);
        ing[s] = true , dis[s] = 0;
        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] <= 0) 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 = 0;
        while(spfa()){
            int pl = inf , p , k;
            for(p=t;p!=s;p=to[k^1]){
                k = pre[p];
                pl = min(pl,cap[k]);
            }
            for(p=t;p!=s;p=to[k^1]){
                k = pre[p];
                cap[k] -= pl;
                cap[k^1] += 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[1010][1010];
int main()
{
    scanf("%d%d%d",&n,&m,&k);
    for (i=1;i<=n;i++)
    scanf("%d",&a[i]),sum+=a[i];
    for (i=1;i<=m;i++)
    scanf("%d",&b[i]),sum+=b[i];
    for (i=1;i<=k;i++)
    {
        int A,B;
        scanf("%d%d",&A,&B);
        flag[A][B]=1;
    }
    

        t.ini();
        t.link(0,n+m+1,n*m,0);
        t.link(n+m+1,n+m+2,n*m,0);
        for (i=1;i<=n;i++)
        {
            t.link(n+m+1,i,a[i],-n*m);
            t.link(n+m+1,i,n*m,0);
        }
        for (i=1;i<=m;i++)
        {
            t.link(n+i,n+m+2,b[i],-n*m);
            t.link(n+i,n+m+2,n*m,0);
        }
        for (i=1;i<=n;i++)
        for (j=1;j<=m;j++)
        if (flag[i][j]==0)
        t.link(i,n+j,1,1);
        
        int ans=t.run(0,n+m+2,n+m+10).second;
        printf("%d
",ans+sum*n*m);
}