Cod sursa(job #2476073)

Utilizator aZvezdaAtanas Dimitrov aZvezda Data 18 octombrie 2019 00:06:07
Problema Flux maxim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.49 kb
#include <bits/stdc++.h>
using namespace std;
//#pragma GCC optimize ("O3")
//#pragma GCC target ("sse4")
#define endl "\n"
#define pb push_back
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
const ll mod = 1e9 + 7, inf = 1e18;


const ll MAX_N = 1e3 + 10;
struct Edge {
    ll to, cap, f;
    Edge(ll to, ll cap) : to(to), cap(cap), f(0) {}
};

struct Dinic {
    ll s, t, n, m;
    vector<ll> dist, g[MAX_N], id;
    vector<Edge> edges;
    Dinic(ll s, ll t, ll n) : s(s), t(t), n(n) {dist.resize(n); id.resize(n);}
    void addEdge(ll a, ll b, ll cap) {
        g[a].push_back(edges.size());
        edges.emplace_back(b, cap);
        g[b].push_back(edges.size());
        edges.emplace_back(a, 0);
    }
    bool bfs() {
        for(ll i = 0; i < n; i ++) {dist[i] = -1; id[i] = 0;}
        queue<ll> q;
        q.push(s);
        dist[s] = 0;
        while(!q.empty()) {
            ll curr = q.front(); q.pop();
            for(auto id : g[curr]) {
                if(dist[edges[id].to] != -1 || edges[id].cap - edges[id].f < 1) {continue;}
                dist[edges[id].to] = dist[curr] + 1;
                q.push(edges[id].to);
            }
        }
        return dist[t] != -1;
    }
    ll dfs(ll curr, ll pushed) {
        if(pushed == 0 || curr == t) {return pushed;}
        for(; id[curr] < g[curr].size();) {
            int nxt = g[curr][id[curr] ++];
            if(dist[edges[nxt].to] != dist[curr] + 1 || edges[nxt].cap - edges[nxt].f < 1) {continue;}
            ll tr = dfs(edges[nxt].to, min(edges[nxt].cap - edges[nxt].f, pushed));
            if(tr != 0) {
                edges[nxt].f += tr;
                edges[nxt ^ 1].f -= tr;
                return tr;
            }
        }
        return 0;
    }
    ll callDfs() {
        return dfs(s, inf);
    }
    ll maxFlow() {
        ll ret = 0;
        while(bfs()) {
            ll p;
            while(p = callDfs()) {
                ret += p;
            }
        }
        return ret;
    }
};
int main() {
    //ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
    ofstream myfile;
    myfile.open ("maxflow.out");
    ifstream myFileIn;
    myFileIn.open ("maxflow.in");

    ll n, m;
    myFileIn >> n >> m;
    Dinic dc(1, n, n + 1);
    for(ll i = 0; i < m; i ++) {
        ll a, b, cap;
        myFileIn >> a >> b >> cap;
        dc.addEdge(a, b, cap);
    }
    myfile << dc.maxFlow() << endl;
    return 0;
}