题目大意:
在前期市场调查和站址勘测之后,公司得到了一共N个可以作为通讯信号中转站的地址,而由于这些地址的地理位置差异,在不同的地方建造通讯中转站需要投入的成本也是不一样的,所幸在前期调查之后这些都是已知数据:建立第i个通讯中转站需要的成本为Pi(1≤i≤N)。另外公司调查得出了所有期望中的用户群,一共M个。关于第i个用户群的信息概括为Ai, Bi和Ci:这些用户会使用中转站Ai和中转站Bi进行通讯,公司可以获益Ci。(1≤i≤M, 1≤Ai, Bi≤N) THU集团的CS&T公司可以有选择的建立一些中转站(投入成本),为一些用户提供服务并获得收益(获益之和)。那么如何选择最终建立的中转站才能让公司的净获利最大呢?(净获利 = 获益之和 - 投入成本之和)
思路:
最小割问题,把所有客服的钱加起来,这样构建一个网络流,s点连到中转站,客户连到t点,中转站和客户之间按要求连边。
程序:
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<queue>
#include<cstring>
#include<algorithm>
#define maxn 50005
using namespace std;
int n,m,cnt,s,t,dep[maxn],head[maxn],cur[maxn];
struct data{int to,next,w;}e[maxn*50];
const int inf=1e9;
queue<int> q;
int add(int x,int y, int w){
e[++cnt].w=w; e[cnt].to=y; e[cnt].next=head[x]; head[x]=cnt;
e[++cnt].w=0; e[cnt].to=x; e[cnt].next=head[y]; head[y]=cnt;
}
int read()
{
int x=0,f=1;char ch=getchar();
while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
bool bfs(){
for (int i=s;i<=t;i++) dep[i]=0;
dep[s]=1;
while (!q.empty()) q.pop();
q.push(s);
while (!q.empty()){
int u=q.front();
q.pop();
for (int i=head[u];i;i=e[i].next)
if (e[i].w&&!dep[e[i].to]){
dep[e[i].to]=dep[u]+1;
if (e[i].to==t) return 1;
q.push(e[i].to);
}
}
return 0;
}
int dfs(int x,int maxf){
if (x==t||!maxf) return maxf;
int ret=0;
for (int &i=cur[x];i;i=e[i].next)
if (e[i].w&&dep[e[i].to]==dep[x]+1){
int f=dfs(e[i].to,min(e[i].w,maxf-ret));
e[i].w-=f;
e[i^1].w+=f;
ret+=f;
if (ret==maxf) break;
}
return ret;
}
int dinic(){
int ans=0;
while (bfs()){
for (int i=s;i<=t;i++) cur[i]=head[i];
ans+=dfs(s,inf);
}
return ans;
}
int main(){
cnt=1;
n=read();m=read();
s=0; t=n+m+1; int ans=0;
for (int i=1;i<=n;i++){
int x=read();
add(i+m,t,x);
}
for (int i=1;i<=m;i++){
int x=read(),y=read(),z=read();
ans+=z;
add(i,x+m,inf);
add(i,y+m,inf);
add(s,i,z);
}
printf("%d",ans-dinic());
return 0;
}