POJ 3469 Dual Core CPU(最小割-Dinic)

Description
n个任务,两个机器AB,输入n对数字Ai,Bi表示i号任务在A上消耗和在B上的消耗。,然后再给m行,每行a,b,c三个数字,表示如果a任务和b任务不在一个机器上工作的话,需要额外花费c。问所有任务都工作的情况下的最小花费
Input
第一行两个整数N和M表示任务数和约束数,之后N行第i行两个整数Ai和Bi分别表示i任务在机器A和B上的消耗,最后M行每行三个整数a,b,c,表示如果a任务和b任务不在一个机器上工作的话,需要额外花费c
Output
输出所有任务都工作的情况下最小花费
Sample Input
3 1
1 10
2 10
10 3
2 3 1000
Sample Output
13
Solution
此题题意可以转化为以最小花费将N个任务分成A和B两部分,正好为最小割的定义,用N个点表示这N个任务,源点到每个任务建一条容量为1的边,每个任务向汇点建一条容量为1的边,对于有约束条件的两个点a和b,建一条容量为c的双向边,最后用dinic算法求出最小割即为答案
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
#define maxn 22222
#define maxm 2222222
#define INF 0x3f3f3f3f
int head[maxn],cur[maxn],d[maxn],st[maxm],s,e,no,n;
struct point
{
    int u,v,flow,next;
    point(){};
    point(int x,int y,int z,int w):u(x),v(y),next(z),flow(w){};
}p[maxm];
void add(int x,int y,int z)
{
    p[no]=point(x,y,head[x],z); 
    head[x]=no++;
    p[no]=point(y,x,head[y],0); 
    head[y]=no++;
}
void init()
{
    memset(head,-1,sizeof(head));
    no=0;
}
bool bfs()
{
    int i,x,y;
    queue<int>q;
    memset(d,-1,sizeof(d));
    d[s]=0; 
    q.push(s);
    while(!q.empty())
    {
        x=q.front();    
        q.pop();
        for(i=head[x];i!=-1;i=p[i].next)
        {
            if(p[i].flow&& d[y = p[i].v]<0)
            {
                d[y]=d[x]+1;
                if(y==e)    
                    return true;
                q.push(y);
            }
        }
    }
    return false;
}
int dinic()
{
    int i,loc,top,x=s,nowflow,maxflow=0;
    while(bfs()){
        for(i=s;i<=e;i++)   
            cur[i]=head[i];
        top=0;
        while(true)
        {
            if(x==e)
            {
                nowflow=INF;
                for(i=0;i<top;i++)
                {
                    if(nowflow>p[st[i]].flow)
                    {
                        nowflow=p[st[i]].flow;
                        loc=i;
                    }
                }
                for(i=0;i<top;i++)
                {
                    p[st[i]].flow-=nowflow;
                    p[st[i]^1].flow+=nowflow;
                }
                maxflow+=nowflow;
                top=loc;    
                x=p[st[top]].u;
            }
            for(i=cur[x];i!=-1;i=p[i].next)
                if(p[i].flow&&d[p[i].v]==d[x]+1) 
                    break;
            cur[x]=i;
            if(i!=-1)
            {
                st[top++]=i;
                x=p[i].v;
            }
            else 
            {
                if(!top)    
                    break;
                d[x]=-1;
                x=p[st[--top]].u;
            }
        }
    }
    return maxflow;
}
int N,M;
int main()
{
    while(~scanf("%d%d",&N,&M))
    {
        init();//初始化 
        s=0;//源点为0 
        e=N+1;//汇点为N+1 
        for(int i=1;i<=N;i++)
        {
            int ai,bi;
            scanf("%d%d",&ai,&bi);
            add(s,i,ai);//源点向每个点建容量为1的边 
            add(i,e,bi);//每个点向汇点建容量为1的边 
        }
        while(M--)
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            add(a,b,c);//a到b建容量为c的边 
            add(b,a,c);//b到a建容量为c的边 
        }
        printf("%d\n",dinic());
    }
    return 0;
} 

你可能感兴趣的:(POJ 3469 Dual Core CPU(最小割-Dinic))