費用流mcmf模板2.0版

標籤：acm   c語言   編程   演算法   網路流   

題目地址：HDU 3488

今晚才發現以前的費用流模板居然是錯的。。。。。。。有漏洞。。。。但是居然都AC了那麼多題。。想想也是。。做的費用流也不多，而且都是流量為1的，這個漏洞是不會影響的。因為以前在每次最小費用增廣後找到的最小流量是經過的所有路的最小流量，不一定是可以到達匯點的最小流量。。但是如果流量都為1的話。。那隻要邊沒有全部流過，那肯定最小流量會是1.而如果已經全部流過的話。。那也沒意義了。。已經到達了最大流了。所以說這個漏洞一直都沒被發現。。。

那麼應該如何改進呢。那就是跟最短路的方法一樣，在標記一個費用數組，隨著路徑的變化而變化。這樣的話，最終的

f[sink]就一定是到達匯點的最小流量。其他地方都是正確的。改了這個地方就OK了。

為了避免前面的代碼禍害人間。。。還是找個時間全改過來吧。。TUT

模板如下；

#include <iostream>#include <stdio.h>#include <string.h>#include <stdlib.h>#include <math.h>#include <ctype.h>#include <queue>#include <map>#include <algorithm>using namespace std;const int INF=0x3f3f3f3f;int head[500], source, sink, cnt, flow, cost;int d[500], pre[500], q[1000000], cur[500], vis[500], f[500];struct node{    int u, v, cap, cost, next;}edge[1000000];void add(int u, int v, int cap, int cost){    edge[cnt].v=v;    edge[cnt].cap=cap;    edge[cnt].cost=cost;    edge[cnt].next=head[u];    head[u]=cnt++;    edge[cnt].v=u;    edge[cnt].cap=0;    edge[cnt].cost=-cost;    edge[cnt].next=head[v];    head[v]=cnt++;}int spfa(){    memset(d,INF,sizeof(d));    memset(vis,0,sizeof(vis));    int minflow=INF, f1=0, f2=0, i;    q[f1++]=source;    f[source]=INF;    d[source]=0;    cur[source]=-1;    while(f1>=f2)    {        int u=q[f2++];        vis[u]=0;        for(i=head[u];i!=-1;i=edge[i].next)        {            int v=edge[i].v;            if(d[v]>d[u]+edge[i].cost&&edge[i].cap)            {                d[v]=d[u]+edge[i].cost;                f[v]=min(f[u],edge[i].cap);                cur[v]=i;                if(!vis[v])                {                    q[f1++]=v;                    vis[v]=1;                }            }        }    }    if(d[sink]==INF) return 0;    flow+=f[sink];    cost+=f[sink]*d[sink];    for(i=cur[sink];i!=-1;i=cur[edge[i^1].v])    {        edge[i].cap-=f[sink];        edge[i^1].cap+=f[sink];    }    return 1;}void mcmf(){    flow=cost=0;    while(spfa()) ;    printf("%d\n",cost);}int main(){    int T, n, m, i, j, a, b, c;    scanf("%d",&T);    while(T--)    {        scanf("%d%d",&n,&m);        memset(head,-1,sizeof(head));        cnt=0;        source=0;        sink=2*n+1;        flow=0;        cost=0;        for(i=1;i<=n;i++)        {            add(source,i,1,0);            add(i+n,sink,1,0);        }        while(m--)        {            scanf("%d%d%d",&a,&b,&c);            add(a,b+n,1,c);        }        mcmf();    }    return 0;}