1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97
| #include <queue> #include <cstdio> #include <cstring> #include <iostream> #include <algorithm> #define int long long using namespace std; const int MAXN = 2e5 + 5; const int MAXM = 5e5 + 5; const int INF = 1e18;
int n, m, sum, ans, tmp, nowst, nowed, balance[MAXN], d[MAXN], cur[MAXN], base[MAXM]; int st, ed, tot = 1, head[MAXN], edge[MAXM], nxt[MAXM], ver[MAXM]; queue<int> q;
void AddEdge(int u, int v, int c) { ver[++tot] = v, edge[tot] = c, nxt[tot] = head[u], head[u] = tot; ver[++tot] = u, edge[tot] = 0, nxt[tot] = head[v], head[v] = tot; }
void Connect(int u, int v, int lower, int upper) { AddEdge(u, v, upper - lower); balance[u] += lower, balance[v] -= lower; }
bool bfs(int st) { while (!q.empty()) q.pop(); memset(d, 0, sizeof(d)); d[st] = 1, q.push(st), cur[st] = head[st];
while (!q.empty()) { int u = q.front(); q.pop(); for (int i = head[u]; i; i = nxt[i]) { int v = ver[i]; if (!edge[i] || d[v]) continue; d[v] = d[u] + 1, cur[v] = head[v]; q.push(v); if (v == ed) return true; } }
return false;
}
int dinic(int u, int flow) {
if (u == ed) return flow;
int res = flow; for (int i = cur[u]; i && res; i = nxt[i]) { int v = ver[i]; if (!edge[i] || d[v] != d[u] + 1) continue; int k = dinic(v, min(res, edge[i])); if (!k) d[v] = 0; edge[i] -= k, edge[i ^ 1] += k, res -= k; cur[u] = i; }
return flow - res;
}
signed main() {
scanf("%lld %lld %lld %lld", &n, &m, &nowst, &nowed); st = n + 1, ed = n + 2; for (int i = 1, u, v, lower, upper; i <= m; i++) { scanf("%lld %lld %lld %lld", &u, &v, &lower, &upper); Connect(u, v, lower, upper); }
for (int i = 1; i <= n; i++) { if (balance[i] > 0) { sum += balance[i]; AddEdge(i, ed, balance[i]); } else { AddEdge(st, i, -balance[i]); } }
while (bfs(st)) while ((tmp = dinic(st, INF))) ans += tmp; Connect(nowed, nowst, 0, INF); while (bfs(st)) while ((tmp = dinic(st, INF))) ans += tmp;
if (ans < sum) { printf("please go home to sleep\n"); } else { printf("%d\n", edge[tot]); }
return 0; }
|