HDU 3879 Base Station（最小割---最大權閉合）

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

題目地址：HDU 3879

無語。。。對這題的資料範圍無語。。剛上來一看，建圖思路很快出來了，但是一看資料範圍。。需要5w個點。。於是我以為需要縮點或是別的最佳化。。於是又想了會怎麼最佳化，感覺沒法最佳化了。。於是上網一搜。。還真都是就這麼過了。。。過了。。過了。。於是，我就按剛上來那個思路敲，結果203ms過了。。。我真懷疑出題者是不是不小心把資料範圍多寫了幾個0.這要放在比賽中。。肯定不敢這麼寫。。。

建圖思路是，對所有任務額外建一個點，並連源點，權值為任務的獲利。然後將每個任務所對應的網站連邊，權值為INF，再將所有網站與匯點連邊，權值為各網站所需要的花費。這其實就是一個最大權閉合模型。

代碼如下：

#include <iostream>#include <cstdio>#include <string>#include <cstring>#include <stdlib.h>#include <math.h>#include <ctype.h>#include <queue>#include <map>#include <set>#include <algorithm>using namespace std;const int INF=0x3f3f3f3f;int head[60000], cnt, source, sink, nv;int d[60000], num[60000], pre[60000], cur[60000];struct node{    int u, v, cap, next;} edge[1000000];void add(int u, int v, int cap){    edge[cnt].v=v;    edge[cnt].cap=cap;    edge[cnt].next=head[u];    head[u]=cnt++;    edge[cnt].v=u;    edge[cnt].cap=0;    edge[cnt].next=head[v];    head[v]=cnt++;}void bfs(){    memset(d,-1,sizeof(d));    memset(num,0,sizeof(num));    queue<int>q;    q.push(sink);    d[sink]=0;    num[0]=1;    while(!q.empty())    {        int u=q.front();        q.pop();        for(int i=head[u]; i!=-1; i=edge[i].next)        {            int v=edge[i].v;            if(d[v]==-1)            {                d[v]=d[u]+1;                num[d[v]]++;                q.push(v);            }        }    }}int isap(){    memcpy(cur,head,sizeof(cur));    int flow=0, u=pre[source]=source, i;    bfs();    while(d[source]<nv)    {        if(u==sink)        {            int f=INF, pos;            for(i=source; i!=sink; i=edge[cur[i]].v)            {                if(f>edge[cur[i]].cap)                {                    f=edge[cur[i]].cap;                    pos=i;                }            }            for(i=source; i!=sink; i=edge[cur[i]].v)            {                edge[cur[i]].cap-=f;                edge[cur[i]^1].cap+=f;            }            flow+=f;            u=pos;        }        for(i=cur[u]; i!=-1; i=edge[i].next)        {            if(d[edge[i].v]+1==d[u]&&edge[i].cap)            {                break;            }        }        if(i!=-1)        {            cur[u]=i;            pre[edge[i].v]=u;            u=edge[i].v;        }        else        {            if(--num[d[u]]==0) break;            int mind=nv;            for(i=head[u]; i!=-1; i=edge[i].next)            {                if(mind>d[edge[i].v]&&edge[i].cap)                {                    mind=d[edge[i].v];                    cur[u]=i;                }            }            d[u]=mind+1;            num[d[u]]++;            u=pre[u];        }    }    return flow;}int main(){    int n, m, a, b, c, i, sum;    while(scanf("%d%d",&n,&m)!=EOF)    {        memset(head,-1,sizeof(head));        cnt=0;        sum=0;        source=0;        sink=n+m+1;        nv=sink+1;        for(i=1;i<=n;i++)        {            scanf("%d",&a);            add(i+m,sink,a);        }        for(i=1;i<=m;i++)        {            scanf("%d%d%d",&a,&b,&c);            add(source,i,c);            add(i,a+m,INF);            add(i,b+m,INF);            sum+=c;        }        printf("%d\n",sum-isap());    }    return 0;}

HDU 3879 Base Station（最小割---最大權閉合）