最大权闭合图
学了一下最大权闭合图。
把所有正权点和源点连边,容量为其点权。把负权点和汇点连边,容量为其点权的绝对值。再根据题目意思在正权点和负权点之间连容量为INF的边。可以证明答案为正权点的和-源点到汇点的最大流。
这道题把边看成一个点,点权为其收益。再和它本来连的两个点各连两条边就好了。
代码:
#include<queue>
#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define M 50005
#define N 55005
#define F inline
using namespace std;
struct edge{ int nxt,to,v,f; }ed[M<<3];
int n,m,k=1,s,t,ti,ans,d[N],h[N],tmp[N],f[N];
queue<int> q;
F char readc(){
static char buf[100000],*l=buf,*r=buf;
if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
return l==r?EOF:*l++;
}
F int _read(){
int x=0; char ch=readc();
while (!isdigit(ch)) ch=readc();
while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=readc();
return x;
}
F bool bfs(){
while (!q.empty()) q.pop();
d[s]=0,f[s]=++ti,q.push(s);
while (!q.empty()){
int x=q.front(); q.pop();
for (int i=h[x],v;i;i=ed[i].nxt)
if (ed[i].v>ed[i].f&&f[v=ed[i].to]!=ti)
d[v]=d[x]+1,f[v]=ti,q.push(v);
}
return f[t]==ti;
}
int dfs(int x,int rem){
if (x==t||rem==0) return rem; int sum=0;
for (int &i=tmp[x];i;i=ed[i].nxt)
if (d[ed[i].to]==d[x]+1){
int p=dfs(ed[i].to,min(rem,ed[i].v-ed[i].f));
if (p) sum+=p,ed[i].f+=p,ed[i^1].f-=p,rem-=p;
if (!rem) break;
}
return sum;
}
F void addedge(int x,int y,int z){
ed[++k]=(edge){h[x],y,z},h[x]=k;
ed[++k]=(edge){h[y],x,0},h[y]=k;
}
int main(){
n=_read(),m=_read(),t=n+m+1;
for (int i=1;i<=n;i++) addedge(i,t,_read());
for (int i=n+1;i<t;i++){
int x=_read(),y=_read(),z=_read();
addedge(i,x,1e9),addedge(i,y,1e9);
addedge(s,i,z),ans+=z;
}
while (bfs()) memcpy(tmp,h,sizeof(h)),ans-=dfs(s,1e9);
return printf("%d\n",ans),0;
}