POJ Dual Core CPU (网络流,最小割)

题目:

   给出由核A和核B组成的双核CPU上的运行N个模块。模块i在核A上执行的花费为Ai,在核B上执行的花费为Bi。有M个互相之间需要进行数据交换的模块组合(ai,bi),如果,这两块模块在同一个核上执行则没有额外的费用,否则会产生wi的花费。请计算执行所有模块所需的最小花费。

思路:

遇到有两个阵营的时候。我们就要想到最小割。

S 连一个阵营,,T 连一个阵营,如果割断一条边,那么他就“叛逃”到了另外一个阵营。

中间两个相互连边,这个值额外的费用。这个为了使 S  T 可以连接起来。。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>
using namespace std;
const int N = 1000000;
const int INF = 0x3f3f3f3f;
#define mem(x,v) memset(x,v,sizeof(x))
struct node{
    int v,next,w;
}f[N];
int cnt,head[N],cur[N],dep[N],s,t,n,m,k;
void Add_edge(int x, int y, int z){
    cnt++;
    f[cnt].v = y;
    f[cnt].w = z;
    f[cnt].next = head[x];
    head[x] = cnt;
}
void init(){
    scanf("%d%d",&n,&m);
    int F,D,z;
    cnt = -1;
    mem(head,-1);
    s = 0; t = n + 1;
    for (int i = 1; i <= n; i++){
        scanf("%d%d",&F,&D);
        Add_edge(i,t,F);
        Add_edge(t,i,0);
        Add_edge(s,i,D);
        Add_edge(i,s,0);
    }
   for (int i = 1; i <= m; i++){
       scanf("%d%d%d",&F,&D,&z);
       Add_edge(F,D,z);
       Add_edge(D,F,0);
       Add_edge(D,F,z);
       Add_edge(F,D,0);
    }
    return;
}
bool bfs(){
    queue<int>q;
    while(!q.empty()) q.pop();
    mem(dep,0);
    dep[s] = 1;
    q.push(s);
    while(!q.empty()){
        int u = q.front(); q.pop();
        for (int i = head[u]; i != -1; i = f[i].next){
            int v = f[i].v;
            if (dep[v] == 0 && f[i].w > 0){
                dep[v] = dep[u] + 1;
                if (v == t) return 1;
                q.push(v);
            }
        }
    }
    if (dep[t] > 0) return 1;
    return 0;
}
int dfs(int u,int flow){
    if (u == t || flow == 0) return flow;
    for (int& i = cur[u]; i != -1; i = f[i].next){
        int v = f[i].v,w = f[i].w;
        if ((dep[v] == dep[u] + 1) && (w > 0)){
            int di = dfs(v,min(flow,w));
            if (di > 0){
                f[i].w -= di;
                f[i^1].w += di;
                return di;
            }
        }
    }
    return 0;
}
int Dinic(){
    int Ans = 0,d;
    while(bfs()){
        for (int i = 0; i <= t; i++)
            cur[i] = head[i];
        while(d = dfs(s,INF)){
            Ans += d;
        }
    }
    return Ans;
}
int main() {
    init();
    printf("%d\n",Dinic());
    return 0;
}

猜你喜欢

转载自blog.csdn.net/kidsummer/article/details/81260987