当前位置: 代码迷 >> 综合 >> 洛谷 P2740 [USACO4.2]草地排水Drainage Ditches (EK增广路算法求最大流模板)
  详细解决方案

洛谷 P2740 [USACO4.2]草地排水Drainage Ditches (EK增广路算法求最大流模板)

热度:65   发布时间:2023-12-06 08:03:57.0

题目:草地排水

思路:EK增广路算法求最大流模板

代码:

#include<bits/stdc++.h>
using namespace std;#define maxn 200
#define read(x) scanf("%d",&x);
#define inf (1<<30)int n,m;
int c[maxn+5][maxn+5];
int f[maxn+5][maxn+5];void readin() {read(m);read(n);for(int i=1;i<=m;i++) {int x,y,z;read(x);read(y);read(z);c[x][y]+=z;}
}queue<int> que,emp;
bool use[maxn+5];
int fa[maxn+5];bool bfs() {que=emp;memset(use,0,sizeof(use));use[1]=true;que.push(1);while(!que.empty()) {int h=que.front();que.pop();for(int i=1;i<=n;i++) {if(use[i]||f[h][i]>=c[h][i]) continue;que.push(i);fa[i]=h;use[i]=true;if(i==n) return true;}}return false;
}vector<int> vec;void count() {vec.clear();int k=n;while(k) {vec.push_back(k);k=fa[k];}
}void change() {count();int s=inf;for(int i=vec.size()-1;i>=1;i--) {int x=vec[i],y=vec[i-1];s=min(s,c[x][y]-f[x][y]);}for(int i=vec.size()-1;i>=1;i--) {int x=vec[i],y=vec[i-1];f[x][y]+=s;c[y][x]+=s;}
}int main() {readin();while(bfs()) {change();}int ans=0;for(int i=1;i<=n;i++) {ans+=f[1][i];}printf("%d",ans);return 0;
}