bzoj 1927 星际竞速（拆点费用流）

题目链接 http://www.lydsy.com/JudgeOnline/problem.php?id=1927

好嘛，现在我要正式用某神犇的博客上的题的顺序开始刷题了。。。

这道题将每个能力爆发模式的费用，即单点费用拆成两个点，再加边。要加入一个源点和汇点。本蒟蒻只会写有点耗时的spfa版，若有那个神犇会zkw费用流，请教一下= =||。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
using namespace std;

#define inf (1e8)
int N,M,S,T;
int a[10000],dep[10000],p[10000],vis[10000],head[10000];

struct node{
	int u,v,w,c,flow,next;
}e[100000];

int k=0;
void adde(int u,int v,int c,int w){
	e[k].u=u; e[k].v=v; e[k].w=w; e[k].c=c;
	e[k].flow=0; e[k].next=head[u]; head[u]=k++;
	
	e[k].u=v; e[k].v=u; e[k].w=-w; e[k].c=0;
	e[k].flow=0; e[k].next=head[v]; head[v]=k++;
}

void input(){
	memset(head,-1,sizeof(head));
	scanf("%d%d",&N,&M);
	S=0; T=2*N+1;
	for(int i=1;i<=N;i++)scanf("%d",&a[i]);
	for(int i=1;i<=N;i++){
		adde(S,i,1,0);
		adde(i+N,T,1,0);
		adde(0,i+N,1,a[i]);
	}
	for(int i=1;i<=M;i++){
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		if(x>y)swap(x,y);
		adde(x,y+N,1,z);
	}
}

void solve(){
	queue<int>q;
	int ans=0;
	while(1){
		for(int i=1;i<=T;i++)dep[i]=inf;
		q.push(S);
		dep[S]=0;
		while(!q.empty()){
			int u=q.front();
			q.pop(); vis[u]=0;
			
			for(int i=head[u];i!=-1;i=e[i].next){
				int v=e[i].v;
				if(dep[v]>dep[u]+e[i].w&&e[i].c>e[i].flow){
					dep[v]=dep[u]+e[i].w;
					p[v]=i;
					if(vis[v]==0){
						vis[v]=1; q.push(v);
					}
				}
			}
		}
		if(dep[T]==inf)break;
		
		int tt=T,a=inf;
		while(tt!=0){
			a=min(a,e[p[tt]].c-e[p[tt]].flow);
			tt=e[p[tt]].u;
		}
		ans+=dep[T]*a;
		tt=T;
		while(tt!=0){
			e[p[tt]].flow+=a;
			e[p[tt]^1].flow-=a;
			tt=e[p[tt]].u;
		}
	}
	printf("%d",ans);
}

int main(){
	input();
	solve();
	return 0;
}