1927: [Sdoi2010]星际竞速

拆点费用流。

每个点拆成出点和入点,S向每个出点连边(v=1,c=0),代表从该点出发可以有一条指向其他点的边。S向每个入点连边(v=1,c=Ai),代表可以以此点为起点。每个入点向T连边(v=1,c=0),代表到达该点。对应边的起点的出点向终点的入点连边(v=1,c=w),代表可以从起点出发到达终点,由此,任意一个入点要么从S出发(代价为Ai)要么从另一点的出点出发(代价为w),符合题意。

然后跑一遍费用流就过了O(∩_∩)O

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int inf=1e9;
struct Edge{int from,to,next,v,c;}e[50005];
int head[1605],d[1605],cnt=1,from[1605];
bool inq[1605];
void ins(int u,int v,int w,int c){
	cnt++;e[cnt].from=u;e[cnt].to=v;e[cnt].next=head[u];head[u]=cnt;e[cnt].v=w;e[cnt].c=c;
}
void insert(int u,int v,int w,int c){
	ins(u,v,w,c);ins(v,u,0,-c);
}
bool spfa(int s,int t,int &cost){
	memset(d,0x3f,sizeof(d));
	d[s]=0;queue<int>q;q.push(s);
	while(!q.empty()){
		int u=q.front();q.pop();inq[u]=false;
		for(int i=head[u];i;i=e[i].next)
		if(e[i].v&&d[e[i].to]>d[u]+e[i].c){
			d[e[i].to]=d[u]+e[i].c;
			from[e[i].to]=i;
			if(!inq[e[i].to]){inq[e[i].to]=true;q.push(e[i].to);}
		}
	}
	if(d[t]>=inf)return false;
	int x=inf;
	for(int i=from[t];i;i=from[e[i].from])x=min(x,e[i].v);
	for(int i=from[t];i;i=from[e[i].from]){
		e[i].v-=x;e[i^1].v+=x;cost+=x*e[i].c;
	}
	return true;
}
int mcmf(int s,int t){int cost=0;while(spfa(s,t,cost));return cost;}
int main(){
	int n,m,u,v,w;scanf("%d%d",&n,&m);
	int S=0,T=n+n+1;
	for(int i=1;i<=n;i++){
		scanf("%d",&w); 
		insert(S,i,1,0);
		insert(S,i+n,1,w);
		insert(i+n,T,1,0);
	}
	while(m--){
		scanf("%d%d%d",&u,&v,&w);
		if(u>v)swap(u,v);
		insert(u,v+n,1,w);
	}
	printf("%d",mcmf(S,T));
	return 0;
}


你可能感兴趣的:(1927: [Sdoi2010]星际竞速)