BZOJ 1497 最大权闭合图

题解:

边由被限制节点指向限制节点

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdlib>
#include <cstdio>

#define N 120000
#define M 2000100
#define INF 1e9

using namespace std;

int head[N],next[M],to[M],len[M];
int layer[N],q[M];
int n,m,cnt,S,T,sum;

inline void add(int u,int v,int w)
{
    to[cnt]=v; len[cnt]=w; next[cnt]=head[u]; head[u]=cnt++;
    to[cnt]=u; len[cnt]=0; next[cnt]=head[v]; head[v]=cnt++;
}

inline void read()
{
    memset(head,-1,sizeof head); cnt=0;
    S=0; T=n+m+1; sum=0;
    for(int i=1,a;i<=n;i++)
    {
        scanf("%d",&a);
        add(i+m,T,a);
    }
    for(int i=1,a,b,c;i<=m;i++)
    {
        scanf("%d%d%d",&a,&b,&c);
        add(S,i,c); add(i,a+m,INF); add(i,b+m,INF);
        sum+=c;
    }
}

inline bool bfs()
{
    memset(layer,-1,sizeof layer);
    int h=1,t=2,sta;
    q[1]=S; layer[S]=0;
    while(h<t)
    {
        sta=q[h++];
        for(int i=head[sta];~i;i=next[i])
            if(len[i]&&layer[to[i]]<0)
            {
                layer[to[i]]=layer[sta]+1;
                q[t++]=to[i];
            }
    }
    return layer[T]!=-1;
}

inline int find(int u,int cur_flow)
{
    if(u==T) return cur_flow;
    int res=0,tmp;
    for(int i=head[u];~i&&res<cur_flow;i=next[i])
        if(len[i]&&layer[to[i]]==layer[u]+1)
        {
            tmp=find(to[i],min(cur_flow-res,len[i]));
            len[i]-=tmp; len[i^1]+=tmp; res+=tmp;
        }
    if(!res) layer[u]=-1;
    return res;
}

inline void go()
{
    int ans=0;
    while(bfs()) ans+=find(S,INF);
    printf("%d\n",sum-ans);
}

int main()
{
    while(scanf("%d%d",&n,&m)!=EOF) read(),go();
    return 0;
}