蓝桥杯 算法训练 网络流裸题(EK算法)

问题描述

一个有向图，求1到N的最大流

输入格式

第一行N M，表示点数与边数
接下来M行每行s t c表示一条从s到t的容量为c的边

输出格式

一个数最大流量

样例输入

6 10
1 2 4
1 3 8
2 3 4
2 4 4
2 5 1
3 4 2
3 5 2
4 6 7
5 4 6
5 6 3

样例输出

8

数据约定：
n<=1000 m<=10000

// 这是一个稠密图(E>VlogV)，使用Dinic算法会超时，这里使用EK算法

#include<bits/stdc++.h>
using namespace std;
const int N=1e3+5;
int n,m;
int G[N][N],pre[N],flow[N];
int bfs(int s,int t){
   for(int i=1;i<=n;++i) pre[i]=-1;
   flow[s]=INT_MAX;
   queue<int>q;
   q.push(s);
   while(!q.empty()){
   	int u=q.front();
   	q.pop();
   	for(int v=1;v<=n;++v){
   		if(pre[v]==-1&&G[u][v]>0){
   			pre[v]=u;
   			flow[v]=min(flow[u],G[u][v]);
   			q.push(v);
   		}
   	}
   }
   if(pre[t]==-1) return -1;
   else return flow[t];
}
int EK(int s,int t){
   int delta=0,ret=0;
   while((delta=bfs(s,t))!=-1){
   	ret+=delta;
   	int p=n;
   	while(p!=1){
   		G[pre[p]][p]-=delta;
   		G[p][pre[p]]+=delta;
   		p=pre[p];
   	}
   }
   return ret;
}
void init(){
   memset(G,0,sizeof(G));
}
int main(int argc, char *argv[]){
   init();
   scanf("%d%d",&n,&m);
   for(int i=0;i<m;++i){
   	int u,v,c;scanf("%d%d%d",&u,&v,&c);
   	G[u][v]+=c;
   }
   printf("%d\n",EK(1,n));	
   return 0;
}