首页 > 代码库 > hdu 3879 最大密集子图(点和边均带权)(模板)

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

/*
最大权闭合图,可以用最大密集子图来解速度更快复杂度低
题解:胡伯涛《最小割模型在信息学竞赛中的应用》
点和边均带权的最大密集子图
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 0x3fffffff
struct 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;}

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