Download Speed

#include <bits/stdc++.h>
using namespace std;

const int N = 505;
long long n, m;
vector<vector<long long>> weigh(N, vector<long long>(N, 0));
vector<vector<long long>> adj(N);


long long flow = 0;
vector<long long> parent(N);
long long new_flow;

long long bfs(long long u, long long t, vector<long long>& parent) {
    fill(parent.begin(), parent.end(), -1);
    parent[u] = -9;
    queue<pair<long long, long long>> q;
    const long long INF = LLONG_MAX / 4;
    q.push({u, INF});

    while (!q.empty()) {
        long long cur = q.front().first;
        long long flow = q.front().second;
        q.pop();

        for (int next : adj[cur]) {
            if (parent[next] == -1 && weigh[cur][next] > 0) {
                parent[next] = cur;
                long long new_flow = min(flow, weigh[cur][next]);
                if (next == t) return new_flow;
                q.push({next, new_flow});
            }
        }
    }
    return 0;
}

int maxflow(long long s, long long t) {
    while ((new_flow = bfs(s, t, parent))) {
        flow += new_flow;
        int tempt = t;
        while (tempt != s) {
            int prev = parent[tempt];
            weigh[prev][tempt] -= new_flow;   //forward
            weigh[tempt][prev] += new_flow;   //bacward 
            adj[tempt].push_back(prev);
            tempt = prev;
        }
    }
    return flow;
}

void solve() {
    cin >> n >> m;
    for (int i = 0; i < m; i++) {
        int a, b, c;
        cin >> a >> b >> c;
        adj[a].push_back(b);
        weigh[a][b] += c;
    }
    cout << maxflow(1, n) << endl;
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
 
    solve();
 
    return 0;
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: WRONG ANSWER

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: WRONG ANSWER

Test 12

Verdict: ACCEPTED