bzoj1497 [NOI2006]最大获利 最大权闭合子图

链接

https://www.lydsy.com/JudgeOnline/problem.php?id=1497

思路

最大权闭合子图的裸题
一开始知道是这个最大权闭合子图(虽然我不知道名字)，但是我不理解
所以还是去复习了一遍。
Orz

错误

写读入优化的时候迷迷糊糊的多打了等号

代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
const int N=1e5+7,inf=0x3f3f3f3f;
using namespace std;
int read() {
    int x=0,f=1;char s=getchar();
    for(;s>'9'||s<'0';s=getchar()) if(s=='0') f=-1;
    for(;s>='0'&&s<='9';s=getchar()) x=x*10+s-'0';
    return x*f;
}
int n,m,S,T,dis[N];
struct node {
    int v,nxt,cap;
}e[2000007];
int head[2000007],tot=1;
void add_edge(int u,int v,int cap) {
    e[++tot].v=v;
    e[tot].cap=cap;
    e[tot].nxt=head[u];
    head[u]=tot;
}
void add(int u,int v,int cap) {add_edge(u,v,cap);add_edge(v,u,0);}
queue<int> q;
bool bfs() {
    memset(dis,-1,sizeof(dis));
    q.push(S);
    dis[S]=0;
    while(!q.empty()) {
        int u=q.front();
        q.pop();
        for(int i=head[u];i;i=e[i].nxt) {
            int v=e[i].v;
            if(dis[v]==-1&&e[i].cap) {
                dis[v]=dis[u]+1;
                q.push(v);
            }
        }
    }
    return dis[T]!=-1;
}
int dfs(int u,int f) {
    if(u==T) return f;
    int rest=f;
    for(int i=head[u];i;i=e[i].nxt) {
        int v=e[i].v;
        if(dis[v]==dis[u]+1&&e[i].cap&&rest) {
            int t=dfs(v,min(e[i].cap,rest));
            if(!t) dis[v]=0;
            e[i].cap-=t;
            e[i^1].cap+=t;
            rest-=t;
        }
    }
    return f-rest;
}
int dinic() {
    int ans=0;
    while(bfs()) {
        ans+=dfs(S,inf);
    }
    return ans;
}
int main() {
    n=read(),m=read();
    S=n+m+1,T=n+m+2;
    for(int i=1;i<=n;++i) {
        int x=read();
        add(i+m,T,x);
    }
    int tot=0;
    for(int i=1;i<=m;++i) {
        int x=read(),y=read(),z=read();
        tot+=z;
        add(S,i,z);
        add(i,x+m,inf);
        add(i,y+m,inf);
    }
    int ans=tot-dinic();
    printf("%d\n",ans);
    return 0;
}