题目地址:HDU 3879
无语。。。对这题的数据范围无语。。刚上来一看,建图思路很快出来了,但是一看数据范围。。需要5w个点。。于是我以为需要缩点或是别的优化。。于是又想了会怎么优化,感觉没法优化了。。于是上网一搜。。还真都是就这么过了。。。过了。。过了。。于是,我就按刚上来那个思路敲,结果203ms过了。。。我真怀疑出题者是不是不小心把数据范围多写了几个0.这要放在比赛中。。肯定不敢这么写。。。
建图思路是,对所有任务额外建一个点,并连源点,权值为任务的获利。然后将每个任务所对应的站点连边,权值为INF,再将所有站点与汇点连边,权值为各站点所需要的花费。这其实就是一个最大权闭合模型。
代码如下:
#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <stdlib.h>
#include <math.h>
#include <ctype.h>
#include <queue>
#include <map>
#include <set>
#include <algorithm>
using namespace std;
const int INF=0x3f3f3f3f;
int head[60000], cnt, source, sink, nv;
int d[60000], num[60000], pre[60000], cur[60000];
struct node
{
int u, v, cap, next;
} edge[1000000];
void add(int u, int v, int cap)
{
edge[cnt].v=v;
edge[cnt].cap=cap;
edge[cnt].next=head[u];
head[u]=cnt++;
edge[cnt].v=u;
edge[cnt].cap=0;
edge[cnt].next=head[v];
head[v]=cnt++;
}
void bfs()
{
memset(d,-1,sizeof(d));
memset(num,0,sizeof(num));
queue<int>q;
q.push(sink);
d[sink]=0;
num[0]=1;
while(!q.empty())
{
int u=q.front();
q.pop();
for(int i=head[u]; i!=-1; i=edge[i].next)
{
int v=edge[i].v;
if(d[v]==-1)
{
d[v]=d[u]+1;
num[d[v]]++;
q.push(v);
}
}
}
}
int isap()
{
memcpy(cur,head,sizeof(cur));
int flow=0, u=pre[source]=source, i;
bfs();
while(d[source]<nv)
{
if(u==sink)
{
int f=INF, pos;
for(i=source; i!=sink; i=edge[cur[i]].v)
{
if(f>edge[cur[i]].cap)
{
f=edge[cur[i]].cap;
pos=i;
}
}
for(i=source; i!=sink; i=edge[cur[i]].v)
{
edge[cur[i]].cap-=f;
edge[cur[i]^1].cap+=f;
}
flow+=f;
u=pos;
}
for(i=cur[u]; i!=-1; i=edge[i].next)
{
if(d[edge[i].v]+1==d[u]&&edge[i].cap)
{
break;
}
}
if(i!=-1)
{
cur[u]=i;
pre[edge[i].v]=u;
u=edge[i].v;
}
else
{
if(--num[d[u]]==0) break;
int mind=nv;
for(i=head[u]; i!=-1; i=edge[i].next)
{
if(mind>d[edge[i].v]&&edge[i].cap)
{
mind=d[edge[i].v];
cur[u]=i;
}
}
d[u]=mind+1;
num[d[u]]++;
u=pre[u];
}
}
return flow;
}
int main()
{
int n, m, a, b, c, i, sum;
while(scanf("%d%d",&n,&m)!=EOF)
{
memset(head,-1,sizeof(head));
cnt=0;
sum=0;
source=0;
sink=n+m+1;
nv=sink+1;
for(i=1;i<=n;i++)
{
scanf("%d",&a);
add(i+m,sink,a);
}
for(i=1;i<=m;i++)
{
scanf("%d%d%d",&a,&b,&c);
add(source,i,c);
add(i,a+m,INF);
add(i,b+m,INF);
sum+=c;
}
printf("%d\n",sum-isap());
}
return 0;
}