Poj 3469 Dual Core CPU

http://poj.org/problem?id=3469

题意：有双核CPU，要运行N个模块，模块i在核A上花费是Ai，在B上为Bi。有M个相互交换数据的组合（ai，bi）。如果ai和bi在一块模块上，没有额外花费；如果不在，有额外花费wi。求最小   　　　花费。

题解：这是一个典型的将最小费用的对象分为两个集合，转换为最小割的题。讲N个点分为两个集合，则花费有4部分组成，在A上执行的花费+在B上实行的花费+（ai在源点，bi在汇点）花费+

　　　（ai在汇点，bi在源点）花费->最小割->最大流

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <string>
#include <vector>
#include <list>
#include <map>
#include <queue>
#include <stack>
#include <bitset>
#include <algorithm>
#include <numeric>
#include <functional>
#include <set>
#include <fstream>

using namespace std;

const int maxn=200010;
const int INF=0x7fffffff;

struct edge{
    int to,cap,rev;
};
vector<edge> G[maxn];
int level[maxn];
int iter[maxn];
int A[maxn],B[maxn];
int a[maxn],b[maxn],w[maxn];

void add_edge(int from,int to,int cap)
{
    G[from].push_back((edge){to,cap,(int)G[to].size()});
    G[to].push_back((edge){from,0,(int)G[from].size()-1});
}

void bfs(int s)
{
    memset(level, -1, sizeof(level));
    queue<int> que;
    level[s]=0;
    que.push(s);
    while (!que.empty()) {
        int v=que.front();
        que.pop();
        for (int i=0; i<G[v].size(); i++) {
            edge &e=G[v][i];
            if (e.cap>0&&level[e.to]<0) {
                level[e.to]=level[v]+1;
                que.push(e.to);
            }
        }
    }
}

int dfs(int v,int t,int f)
{
    if (v==t) {
        return f;
    }
    for (int &i=iter[v]; i<G[v].size(); i++) {
        edge &e=G[v][i];
        if (e.cap>0&&level[v]<level[e.to]) {
            int d=dfs(e.to, t, min(f,e.cap));
            if (d>0) {
                e.cap-=d;
                G[e.to][e.rev].cap+=d;
                return d;
            }
        }
    }
    return 0;
}

int max_flow(int s,int t)
{
    int flow=0;
    for (; ; ) {
        bfs(s);
        if (level[t]<0) {
            return flow;
        }
        memset(iter, 0, sizeof(iter));
        int f;
        while ((f=dfs(s, t, INF))>0) {
            flow+=f;
        }
    }
}
int main()
{
    //freopen("/Users/apple/Desktop/POJ 3469/POJ 3469/in", "r", stdin);
    //freopen("/Users/apple/Desktop/POJ 3469/POJ 3469/out", "w", stdout);
    int N,M;
    while ((scanf("%d%d",&N,&M))!=EOF) {
        for (int i=0; i<N; i++) {
            scanf("%d%d",&A[i],&B[i]);
        }
        for (int i=0; i<M; i++) {
            scanf("%d%d%d",&a[i],&b[i],&w[i]);
        }
        int s=N,t=s+1;
        for (int i=0; i<N; i++) {
            add_edge(i, t, A[i]);
            add_edge(s, i, B[i]);
        }
        for (int i=0; i<M; i++) {
            add_edge(a[i]-1, b[i]-1, w[i]);
            add_edge(b[i]-1, a[i]-1, w[i]);
        }
        
        printf("%d
",max_flow(s,t));
    }
    return 0;
}