https://www.lydsy.com/JudgeOnline/problem.php?id=1497
思路:只有建立了中转站
A
i
A_i
Ai和
B
i
B_i
Bi后,才能从用户
i
i
i身上得到
C
i
C_i
Ci收益,这个意思是不是有点像最大权闭合图?
没错就是这个!我们来思考一下怎么建图,因为中转站
i
i
i的代价是
P
i
P_i
Pi,所以我们把每一个中转站与汇点连上权为
P
i
P_i
Pi的边,而用户
j
j
j的收益是
C
j
C_j
Cj,所以从源点向
j
j
j连一条权为
C
j
C_j
Cj的边,同时因为用户
j
j
j需要中转站
A
j
A_j
Aj和
B
j
B_j
Bj,所以从
j
j
j向
A
j
A_j
Aj和
B
j
B_j
Bj各连一条权为
I
N
F
INF
INF的边,然后跑最大流就行了。
#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;
const int maxn=5e3+5;
const int maxm=5e4+5;
const int Max=maxn+maxm;
struct Edge
{
int to,nxt,f;
}edge[6*Max];
int n,m,s,t,tot;
int head[Max],dep[Max],cnt[Max],cur[Max];
inline void addedge(int u,int v,int f)
{
edge[++tot].to=v,edge[tot].nxt=head[u],edge[tot].f=f,head[u]=tot;
edge[++tot].to=u,edge[tot].nxt=head[v],edge[tot].f=0,head[v]=tot;
}
void bfs()
{
memset(dep,-1,sizeof(dep));
queue<int> q;
dep[t]=0,cnt[0]=1;
q.push(t);
while(!q.empty())
{
int u=q.front();
q.pop();
for(int i=head[u];i;i=edge[i].nxt)
{
int v=edge[i].to;
if(!dep[v])
dep[v]=dep[u]+1,++cnt[dep[v]],q.push(v);
}
}
}
int dfs(int u,int lim)
{
if(u==t)
return lim;
int ans=0;
for(int i=cur[u];i;i=edge[i].nxt)
{
cur[u]=i;
int v=edge[i].to;
if(edge[i].f&&dep[u]==dep[v]+1)
{
int flow=dfs(v,min(lim,edge[i].f));
edge[i].f-=flow;
edge[i^1].f+=flow;
lim-=flow;
ans+=flow;
if(!lim)
break;
}
}
if(lim)
{
if(--cnt[dep[u]]==0)
dep[s]=t+1;
++cnt[++dep[u]];
}
return ans;
}
int isap()
{
int ans=0;
bfs();
while(dep[s]<=t)
{
memcpy(cur,head,sizeof(cur));
ans+=dfs(s,INF);
}
return ans;
}
int main()
{
while(~scanf("%d %d",&n,&m))
{
memset(head,0,sizeof(head));
tot=1;
s=0,t=n+m+1;
int u,v,w;
for(int i=1;i<=n;i++)
{
scanf("%d",&u);
addedge(i+m,t,u); //第i个中转站向汇点连权为成本的边
}
int sum=0;
for(int i=1;i<=m;i++)
{
scanf("%d %d %d",&u,&v,&w);
addedge(s,i,w); //源点向第i个客户连权为收益的边
addedge(i,u+m,INF);//第i个用户向两个中转站连权为INF的边
addedge(i,v+m,INF);
sum+=w;
}
printf("%d\n",sum-isap());
}
return 0;
}