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 98 99 100 101 102 103 104 105 106 107 108
| #include<bits/stdc++.h> using namespace std; #define ll long long const int N = 1e5 + 10; const int INF = 0x3f3f3f3f; const int mod = 1e9 + 7; unordered_map<int, int>mp; int k, n, m, S, T; struct E { int v, w; }; vector<E>G[N]; struct mxfl { int n = 0; int m = 0; int s, t; struct Edge { int from, to, cap, flow; }; vector<Edge>edges; vector<int>G[N]; void add_edge(int from, int to, int cap) { edges.push_back(Edge{ from,to,cap,0 }); edges.push_back(Edge{ to,from,0,0 }); m = (int)edges.size(); G[from].push_back(m - 2); G[to].push_back(m - 1); } bool vis[N]; int d[N], cur[N]; bool bfs() { memset(vis, 0, sizeof(vis)); queue<int>Q; Q.push(s); d[s] = 0; vis[s] = 1; while (!Q.empty()) { int x = Q.front(); Q.pop(); for (int i = 0; i < G[x].size(); i++) { Edge& e = edges[G[x][i]]; if (!vis[e.to] && e.cap > e.flow) { vis[e.to] = 1; d[e.to] = d[x] + 1; Q.push(e.to); } } } return vis[t]; } int dfs(int x, int a) { if (x == t || a == 0)return a; int flow = 0, f; for (int& i = cur[x]; i < G[x].size(); i++) { Edge& e = edges[G[x][i]]; if (d[x] + 1 == d[e.to] && (f = dfs(e.to, min(a, e.cap - e.flow))) > 0) { e.flow += f; edges[G[x][i] ^ 1].flow -= f; flow += f; a -= f; if (a == 0)break; } } return flow; } int maxflow() { int flow = 0; while (bfs()) { memset(cur, 0, sizeof(cur)); flow += dfs(s, INF); } return flow; } }MF; int tot, pre; bool check(int t) { for (int i = 1; i <= tot; i++) MF.add_edge(i + (t - 1)*tot, i + t * tot, INF); for (int u = 1; u <= n; u++) { for (E e : G[u]) { int v = e.v, w = e.w; MF.add_edge(u + (t - 1)*tot, v + t * tot, w); } } MF.s = S, MF.t = tot * t + T; int nw = MF.maxflow(); if (pre + nw >= k)return true; pre += nw; return false; } int main() { scanf("%d%d%d%d%d", &k, &n, &m, &S, &T); for (int i = 0; i < m; i++) { int u, v, w; scanf("%d%d%d", &u, &v, &w); G[u].push_back(E{ v,w }); } tot = n; for (int t = 1; t <= 10000; t++) { if (check(t)) { printf("%d\n", t); return 0; } } return 0; }
|