2018.10.10【CQOI2015】【BZOJ3931】【洛谷P3171】网络吞吐量（最短路）（最大流）

BZOJ传送门

洛谷传送门

---

解析：

好粗暴的最短路加最大流。。。

思路：

首先题目要求数据沿最短路传递，而且题目都说了$Dijkstra$，怎么还有人写$SPFA$，不怕被卡吗？

于是，我们先$Dijkstra$求出最短路。

然后根据题目要求构建流网络图，就是只有在最短路树里面的边才能够加入流网络，并且容量设为$INF$。因为题目说了每条路径的容量不限。

然后又是经典套路，点有容量怎么办？拆点啊，中间连点容量大小的边。
直接最大流完事。

---

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define re register
#define gc getchar
#define pc putchar
#define cs const

inline int getint(){
	re int num;
	re char c;
	while(!isdigit(c=gc()));num=c^48;
	while(isdigit(c=gc()))num=(num<<1)+(num<<3)+(c^48);
	return num;
}

cs int N=1003,M=101005;
cs int S=0,T=1002;
cs ll INF=0x3f3f3f3f3f3f3f3f;
int last[N],nxt[M<<1],to[M<<1],ecnt=1;
ll cap[M<<1];
inline void addedge(int u,int v,ll val){
	nxt[++ecnt]=last[u],last[u]=ecnt,to[ecnt]=v,cap[ecnt]=val;
	nxt[++ecnt]=last[v],last[v]=ecnt,to[ecnt]=u,cap[ecnt]=0;
}

int lev[N],cur[N];
inline bool BFS(){
	memset(lev,-1,sizeof lev);
	queue<int> q;
	q.push(S),lev[S]=0,cur[S]=last[S];
	while(!q.empty()){
		int u=q.front();
		q.pop();
		for(int re e=last[u],v=to[e];e;v=to[e=nxt[e]]){
			if(cap[e]&&lev[v]==-1){
				lev[v]=lev[u]+1;
				if(v==T)return true;
				cur[v]=last[v];
				q.push(v);
			}
		}
	}
	return false;
}

inline ll Dinic(cs int &u,cs ll &flow){
	if(u==T)return flow;
	ll ans=0;
	for(int &e=cur[u],v=to[e];e;v=to[e=nxt[e]]){
		if(cap[e]&&lev[v]>lev[u]){
			ll delta=Dinic(v,min(flow-ans,cap[e]));
			if(delta){
				cap[e]-=delta;
				cap[e^1]+=delta;
				ans+=delta;
				if(ans==flow)return ans;
			}
		}
	}
	return ans;
}

inline ll maxflow(){
	ll ans=0;
	while(BFS())ans+=Dinic(S,INF);
	return ans;
}

int c[N];
namespace SP{//Shortest_Path
	int last[N],nxt[M<<1],to[M<<1],ecnt;
	ll w[M<<1];
	inline void Addedge(int u,int v,ll val){
		nxt[++ecnt]=last[u],last[u]=ecnt,to[ecnt]=v,w[ecnt]=val;
		nxt[++ecnt]=last[v],last[v]=ecnt,to[ecnt]=u,w[ecnt]=val;
	}
	
	ll dist[N];
	bitset<N> vis;
	inline void dijkstra(){
		priority_queue<pair<ll,int>,vector<pair<ll,int> > ,greater<pair<ll,int> > > q;
		vis.reset();
		memset(dist,0x3f,sizeof dist);
		q.push(make_pair(dist[1]=0,1));
		while(!q.empty()){
			int u=q.top().second;
			q.pop();
			if(vis[u])continue;
			vis[u]=true;
			for(int re e=last[u],v=to[e];e;v=to[e=nxt[e]])
			if(dist[v]>dist[u]+w[e])q.push(make_pair(dist[v]=dist[u]+w[e],v));
		}
	}
}
using SP::Addedge;
using SP::dist;

int n,m;
signed main(){
	n=getint(),m=getint();
	for(int re i=1;i<=m;++i){
		int u=getint(),v=getint(),d=getint();
		Addedge(u,v,d);
	}
	SP::dijkstra();
	for(int re u=1;u<=n;++u)
	for(int re e=SP::last[u],v=SP::to[e];e;v=SP::to[e=SP::nxt[e]]){
		if(dist[v]==dist[u]+SP::w[e])addedge(u+500,v,INF);
	}
	for(int re i=1;i<=n;++i){
		int val=getint();
		addedge(i,i+500,val);
	}
	addedge(S,1+500,INF);
	addedge(n,T,INF);
	cout<<maxflow();
	return 0;
}