一道最大权闭合子图模板题
好像就是网络流24题中太空飞行计划…
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <queue>
using namespace std;
const int N=1000010,inf=1<<29;
int n,m,S,T,cnt=1,G[N],cur[N];
struct edge{
int t,nx,f;
}E[N<<1];
inline void link(int x,int y,int z){
E[++cnt].t=y; E[cnt].nx=G[x]; G[x]=cnt; E[cnt].f=z;
E[++cnt].t=x; E[cnt].nx=G[y]; G[y]=cnt; E[cnt].f=0;
}
int dis[N];
queue<int> Q;
inline bool bfs(){
for(int i=S;i<=T;i++) dis[i]=-1;
while(!Q.empty()) Q.pop();
dis[S]=0; Q.push(S);
while(!Q.empty()){
int x=Q.front(); Q.pop();
for(int i=G[x];i;i=E[i].nx)
if(E[i].f && dis[E[i].t]==-1){
dis[E[i].t]=dis[x]+1; Q.push(E[i].t);
if(E[i].t==T) return true;
}
}
return false;
}
int dfs(int x,int f){
if(x==T || !f) return f;
int used=0,w;
for(int &i=cur[x];i;i=E[i].nx)
if(E[i].f && dis[E[i].t]==dis[x]+1){
w=dfs(E[i].t,min(f-used,E[i].f));
E[i].f-=w; E[i^1].f+=w;
if((used+=w)==f) return f;
}
if(!used) dis[x]=-1;
return used;
}
int main(){
scanf("%d%d",&n,&m);
S=0; T=n+m+1;
for(int i=1,x;i<=n;i++)
scanf("%d",&x),link(S,i,x);
int ans=0;
for(int i=1,a,b,c;i<=m;i++){
scanf("%d%d%d",&a,&b,&c);
ans+=c; link(i+n,T,c);
link(a,i+n,inf); link(b,i+n,inf);
}
while(bfs()){
for(int i=S;i<=T;i++) cur[i]=G[i];
ans-=dfs(S,inf);
}
printf("%d\n",ans);
return 0;
}