最大流入门题,不用想就知道用Dinic算法。
0ms稳!
#include <cstdio>
#include <cstring>
#include <vector>
#include <queue>
using namespace std;
const int maxn = 200 + 10;
const int INF = 10000000 + 10;
struct edge {
int to, cap, rev;
};
vector<edge> G[maxn];
int level[maxn];
int iter[maxn];
void add_edge(int from, int to, int cap) {
G[from].push_back((edge){to, cap, G[to].size()});
G[to].push_back((edge){from, 0, G[from].size() - 1});
}
void Bfs(int s) {
memset(level, -1, sizeof(level));
queue<int> que;
level[s] = 0;
que.push(s);
while (!que.empty()) {
int v = que.front();
que.pop();
for (int i = 0; i < G[v].size(); i++) {
edge &e = G[v][i];
if (e.cap > 0 && level[e.to] < 0) {
level[e.to] = level[v] + 1;
que.push(e.to);
}
}
}
}
int Dfs(int v, int t, int f) {
if (v == t) {
return f;
}
for (int &i = iter[v]; i < G[v].size(); i++) {
edge &e = G[v][i];
if (e.cap > 0 && level[v] < level[e.to]) {
int d = Dfs(e.to, t, min(f, e.cap));
if (d > 0) {
e.cap -= d;
G[e.to][e.rev].cap += d;
return d;
}
}
}
return 0;
}
int max_flow(int s, int t) {
int flow = 0;
for ( ; ; ) {
Bfs(s);
if (level[t] < 0) {
return flow;
}
memset(iter, 0, sizeof(iter));
int f;
while ((f = Dfs(s, t, INF)) > 0) {
flow += f;
}
}
}
int main(int argc, char const *argv[]) {
int m, n;
while (scanf("%d%d", &m, &n) == 2) {
for (int i = 0; i < m; i++) {
int from, to, cap;
scanf("%d%d%d", &from, &to, &cap);
add_edge(from, to, cap);
}
printf("%d\n", max_flow(1, n));
for (int i = 1; i <= n; i++) {
G[i].clear();
}
}
return 0;
}