Cod sursa(job #1988649)

Utilizator flaviu_2001Craciun Ioan-Flaviu flaviu_2001 Data 3 iunie 2017 23:25:07
Problema Flux maxim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.66 kb
#include <fstream>
#include <vector>
#include <cstring>
#include <queue>

using namespace std;

vector<int> v[1005];
int c[1005][1005], f[1005][1005], n, m, vis[1005], tt[1005];

bool bfs()
{
    memset(vis, 0, sizeof(vis));
    queue<int> q;
    q.push(1);
    vis[1] = 1;
    while(!q.empty()){
        int nod = q.front();
        q.pop();
        if(nod == n)
            continue;
        for (unsigned j = 0; j < v[nod].size(); ++j){
            int x = v[nod][j];
            if(c[nod][x] == f[nod][x] || vis[x])
                continue;
            vis[x] = 1;
            q.push(x);
            tt[x] = nod;
            if(x == n)
                return 1;
        }
    }
    return 0;
}

int abs(int x)
{
    return x >= 0 ? x : -x;
}

int main()
{
    ifstream fin ("maxflow.in");
    ofstream fout ("maxflow.out");
    fin >> n >> m;
    while(m--){
        int x, y, z;
        fin >> x >> y >> z;
        v[x].push_back(y);
        v[y].push_back(x);
        c[x][y] = z;
    }
    int flow, fmin;

    for (flow = 0; bfs();)
        for (unsigned i = 0; i < v[n].size(); ++i){
            int nod = v[n][i];
            if(f[nod][n] == c[nod][n] || !vis[nod])
                continue;
            tt[n] = nod;
            fmin = 1000000000;
            for (int nod = n; nod != 1; nod = tt[nod])
                fmin = min(fmin, c[tt[nod]][nod] - f[tt[nod]][nod]);
            if(fmin == 0)
                continue;
            for (int nod = n; nod != 1; nod = tt[nod]){
                f[tt[nod]][nod] += fmin;
                f[nod][tt[nod]] -= fmin;
            }
            flow += fmin;
        }
    fout << flow << "\n";
    fin.close();
    fout.close();
    return 0;
}