天天看點

BZOJ1497 [NOI2006]最大獲利(洛谷P4174)最大權閉合圖

最大權閉合圖

BZOJ題目傳送門

洛谷題目傳送門

學了一下最大權閉合圖。

把所有正權點和源點連邊,容量為其點權。把負權點和彙點連邊,容量為其點權的絕對值。再根據題目意思在正權點和負權點之間連容量為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;
}