BZOJ 3931 网络吞吐量 （最短路+最大流）

题目大意：

中文题，题目即题意。

解题思路：

先求最短路，只有在最短路上的边才是在网络里的。

然后把这些边上的点拆边。

求最大流。

学习到了优先队列。

（不太会）

第一次WA是挂在我自己宏定义那个max上……

#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<set>
#include<queue>
#define LL long long
#define db double
#define maxn 10000000
#define EPS 1e-15
//#define max(a,b) ((a>b)?(a):(b))
//#define min(a,b) ((a<b)?(a):(b))

#define pa pair<int,int>
#define inf 100000000000000LL
using namespace std;

int n,m,cnt,T;
LL dis[505];
int last[1005],cur[1005],q[1005],h[1005];
int a[100005],b[100005],d[100005];
struct edge{
	int to,next; LL w;
}e[200005];
void insert(int u,int v,LL w){
	e[++cnt].to=v;
	e[cnt].next=last[u];
	last[u]=cnt;
	e[cnt].w=w;
}
void dijkstra(){
	priority_queue<pa,vector<pa>,greater<pa> >q;
	for(int i=1;i<=n;i++)dis[i]=inf;
	dis[1]=0;q.push(make_pair(0,1));
	while(!q.empty())
	{
		int now=q.top().second;q.pop();
		for(int i=last[now];i;i=e[i].next)
			if(dis[now]+e[i].w<dis[e[i].to])
			{
				dis[e[i].to]=dis[now]+e[i].w;
				q.push(make_pair(dis[e[i].to],e[i].to));
			}
	}
}
bool dinic_bfs(){
	int head=0,tail=1;
	for (int i=1;i<=2*n;i++) h[i]=-1;
	q[0]=1; h[1]=0;
	while (head!=tail){
		int x=q[head]; head++;
		for (int i=last[x];i;i=e[i].next)
			if (e[i].w && h[e[i].to]==-1){
				h[e[i].to]=h[x]+1;
				q[tail++]=e[i].to;
			}
	}
	return h[2*n]!=-1;
}
LL dinic_dfs(int x,LL f){
	if (x==2*n) return f;
	LL w,used=0;
	for (int i=cur[x];i;i=e[i].next)
		if (h[e[i].to]==h[x]+1){
			w=dinic_dfs(e[i].to,min(f-used,e[i].w));
			e[i].w-=w;
			e[i^1].w+=w;
			used+=w; if (used==f) return f;
		}
	if (!used) h[x]=-1;
	return used;
}
LL dinic_flow(){
	LL ans=0;
	while (dinic_bfs()){
		for (int i=1;i<=2*n;i++) cur[i]=last[i];
		ans+=dinic_dfs(1,inf);
	}
	return ans;
}
int main(){
	scanf("%d%d",&n,&m);
	for (int i=1;i<=m;i++){
		scanf("%d%d%d",&a[i],&b[i],&d[i]);
		insert(a[i],b[i],d[i]);
		insert(b[i],a[i],d[i]);
	}
	dijkstra();
	for(int i=1;i<=n;i++)last[i]=0;
	cnt=1;
	for(int i=1;i<=m;i++){
		if(dis[a[i]]+d[i]==dis[b[i]])
		{
			insert(a[i]+n,b[i],inf);
			insert(b[i],a[i]+n,0);
		}
		if(dis[b[i]]+d[i]==dis[a[i]])
		{
			insert(b[i]+n,a[i],inf);
			insert(a[i],b[i]+n,0);
		}
	}
	int c;
	for(int i=1;i<=n;i++){
		scanf("%d",&c);
		if(i!=1 && i!=n)insert(i,i+n,c);
		else insert(i,i+n,inf);
		insert(i+n,i,0);
	}
	printf("%lld\n",dinic_flow());
	return 0;
}