hdu 3879 最大密集子图（点和边均带权）（模板）

来源：互联网发布：用java实现四则运算编辑：程序博客网时间：2024/05/21 06:52
/*最大权闭合图，可以用最大密集子图来解速度更快复杂度低题解：胡伯涛《最小割模型在信息学竞赛中的应用》点和边均带权的最大密集子图s-i,权为U=点权绝对值和+边的所有权值i-t，权为U+点的值-点的度u-v，权值为w，意思是选了v后可以获利多少最大获利=（U*n-flow）/2;*/#include<stdio.h>#include<string.h>#include<math.h>#include<queue>#include<algorithm>using namespace std;#define N 5100#define NN 51000#define inf 0x3fffffffstruct node {int u,v,w,next;}bian[NN*8];int p[N],degree[N];int head[N],yong,dis[N],work[N];void init() {yong=0;memset(head,-1,sizeof(head));}void addedge(int u,int v,int w) {bian[yong].v=v;bian[yong].w=w;bian[yong].next=head[u];head[u]=yong++;}int bfs(int s,int t){    memset(dis,-1,sizeof(dis));    queue<int>q;    q.push(s);    dis[s]=0;    while(!q.empty())    {        int u=q.front();        q.pop();        for(int i=head[u];i!=-1;i=bian[i].next)        {            int v=bian[i].v;            if(bian[i].w&&dis[v]==-1)            {                dis[v]=dis[u]+1;                q.push(v);                if(v==t)                    return 1;            }        }    }    return 0;}int dfs(int  s,int limit,int t){    if(s==t)return limit;    for(int &i=work[s];i!=-1;i=bian[i].next)    {        int v=bian[i].v;        if(bian[i].w&&dis[v]==dis[s]+1)        {            int tt=dfs(v,min(limit,bian[i].w),t);            if(tt)            {                bian[i].w-=tt;                bian[i^1].w+=tt;                return tt;            }        }    }    return 0;}int dinic(int s,int t){    int ans=0;    while(bfs(s,t))    {        memcpy(work,head,sizeof(head));        while(int tt=dfs(s,inf,t))            ans+=tt;    }    return ans;}int main() {     int n,m,i,s,t,U,u,v,w;     while(scanf("%d%d",&n,&m)!=EOF) {            U=0;        for(i=1;i<=n;i++) {        scanf("%d",&p[i]);        U+=abs(p[i]);        }        memset(degree,0,sizeof(degree));         init();        for(i=1;i<=m;i++) {            scanf("%d%d%d",&u,&v,&w);            U+=w;            degree[u]+=w;            degree[v]+=w;            addedge(u,v,w);            addedge(v,u,w);        }        s=0;t=n+1;        for(i=1;i<=n;i++) {            addedge(s,i,U);            addedge(i,s,0);            addedge(i,t,U+2*p[i]-degree[i]);            addedge(t,i,0);        }        printf("%d\n",(U*n-dinic(s,t))/2);     }return 0;}
1 0