BZOJ3931 CQOI2015 网络吞吐量
最短路生成图上跑最大流
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
1 #include<iostream> 2 #include<cstdio> 3 #include<algorithm> 4 #include<cstring> 5 #include<cstdlib> 6 #include<cmath> 7 #include<vector> 8 9 using namespace std; 10 11 #define dout printf 12 13 typedef long long ll; 14 const int Maxn=1010; 15 const ll INF=4557430888798830399ll; 16 int n,m,N; 17 ll g[Maxn][Maxn]; 18 ll d[Maxn*2]; 19 bool flag[Maxn]; 20 21 struct Edge{ 22 int to; 23 ll cap,flow; 24 int next; 25 Edge(int to=0,ll cap=0,int next=0):to(to),cap(cap),next(next){ 26 flow=0; 27 } 28 ll adv(){ 29 return cap-flow; 30 } 31 }edges[2000010];int tot=1,fir[Maxn*2]; 32 33 void AddEdge(int from,int to,ll cap){ 34 edges[++tot]=Edge(to,cap,fir[from]);fir[from]=tot; 35 edges[++tot]=Edge(from,0,fir[to]);fir[to]=tot; 36 } 37 38 void init(){ 39 scanf("%d%d",&n,&m);N=2*n; 40 memset(g,0x3f,sizeof g); 41 for(int u,v,w,i=1;i<=m;i++){ 42 scanf("%d%d%d",&u,&v,&w); 43 if(g[u][v]>w) g[u][v]=g[v][u]=w; 44 } 45 46 for(int c,i=1;i<=n;i++){ 47 scanf("%d",&c); 48 AddEdge(i,i+n,(i==1||i==n)?INF:c); 49 } 50 } 51 52 void Dijkstra(){ 53 memset(d,0x3f,sizeof d); 54 memset(flag,0,sizeof flag); 55 d[1]=0; 56 ll MN; 57 int u; 58 for(int T=n;T--;){ 59 MN=INF; 60 for(int i=1;i<=n;i++) if(!flag[i] && d[i]<MN) { 61 MN=d[i]; u=i; 62 } 63 flag[u]=1; 64 for(int i=1;i<=n;i++){ 65 d[i]=min(d[i],d[u]+g[u][i]); 66 } 67 } 68 } 69 70 int q[Maxn],ql,qr; 71 void build(){ 72 q[qr=(ql=0)+1]=n; 73 memset(flag,0,sizeof flag); 74 for(;ql<qr;){ 75 int u=q[++ql]; 76 for(int v=1;v<=n;v++)if(u!=v && d[v]+g[u][v]==d[u]){ 77 AddEdge(v+n,u,INF); 78 if(!flag[v]) q[++qr]=v,flag[v]=1; 79 } 80 } 81 } 82 83 // network-flows begin---------------------------------------------- 84 int s,t; 85 int p[Maxn*2],cur[Maxn*2],num[Maxn*2]; 86 #define e edges[i] 87 88 inline ll Augment(){ 89 ll a=INF; 90 for(int x=t;x!=s;x=edges[p[x]^1].to){ 91 a=min(a,edges[p[x]].adv()); 92 } 93 for(int x=t;x!=s;x=edges[p[x]^1].to){ 94 edges[p[x]].flow+=a; 95 edges[p[x]^1].flow-=a; 96 } 97 return a; 98 } 99 inline void BFS(int start,bool flag){ 100 for(int i=1;i<=N;i++)d[i]=N; 101 d[q[qr=(ql=0)+1]=start]=0; 102 for(int x;ql<qr;){ 103 x=q[++ql]; 104 for(int i=fir[x];i;i=e.next){ 105 if((flag^(bool)e.adv())&&d[e.to]==N){ 106 d[q[++qr]=e.to]=d[x]+1; 107 } 108 } 109 } 110 } 111 inline ll ISAP(){ 112 s=1,t=N; 113 BFS(t,1); 114 ll flow=0; 115 memcpy(cur,fir,sizeof cur); 116 for(int i=1;i<=N;i++)num[d[i]]++; 117 for(int i=1;i<=d[s];i++)if(!num[i])return 0; 118 for(int x=s;d[s]<N;){ 119 if(x==t){ 120 flow+=Augment(); 121 x=s; 122 } 123 int ok=0; 124 for(int&i=cur[x];i;i=e.next){ 125 if(e.adv()&&d[x]==d[e.to]+1){ 126 p[x=e.to]=i; 127 ok=1; 128 break; 129 } 130 } 131 if(!ok){ 132 int M=N; 133 for(int i=fir[x];i;i=e.next){ 134 if(e.adv())M=min(M,(int)d[e.to]+1); 135 } 136 if(!--num[d[x]])break; 137 num[d[x]=M]++; 138 cur[x]=fir[x]; 139 if(x!=s)x=edges[p[x]^1].to; 140 } 141 } 142 return flow; 143 } 144 //network-flows end----------------------------------------------------------- 145 146 int main(){ 147 freopen("network.in","r",stdin); 148 freopen("network.out","w",stdout); 149 init(); 150 Dijkstra(); 151 build(); 152 cout<<ISAP(); 153 return 0; 154 }
原文出处http://www.cnblogs.com/showson/