Cod sursa(job #2328323)

Utilizator mlupseLupse-Turpan Mircea mlupse Data 25 ianuarie 2019 17:13:30
Problema Flux maxim Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.48 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("maxflow.in");
ofstream fout("maxflow.out");
const int NMax = 1000;
const int oo = 2000000000;
vector <int> G[NMax + 5];
int C[NMax + 5][NMax + 5],F[NMax + 5][NMax + 5];
int N,M,Flow;
int Q[NMax + 5], Use[NMax + 5],TT[NMax + 5];
void Read()
{
    fin >> N >> M;
    for(int i = 1; i <= M; ++i)
    {
        int x,y,z;
        fin >> x >> y >> z;
        G[x].push_back(y);
        G[y].push_back(x);
        C[x][y] = z;
        C[y][x] = 0;
    }
}

int BFS()
{
    for(int i = 1; i <= N; ++i)
        Use[i] = TT [i] = 0;
    Q[1] = Q[0] = 1;
    Use[1] = 1;
    for(int i = 1; i <= Q[0]; ++i)
    {
        int Nod = Q[i];
        for(int i = 0; i <(int)G[Nod].size(); ++i)
        {
            int Vecin = G[Nod][i];
            if(Use[Vecin] || C[Nod][Vecin] == F[Nod][Vecin]) continue;
            TT[Vecin] = Nod;
            Q[++Q[0]] = Vecin;
            Use[Vecin] = 1;
            if(Vecin == N) return 1;
        }
    }
    return 0;
}

void Print()
{
    fout << Flow << "\n";
}

int main()
{
    Read();
    while(BFS())
    {
        int Fmin = oo;
        for(int i = N; i != 1; i= TT[i])
        {
            Fmin = min(Fmin,C[TT[i]][i] - F[TT[i]][i]);
        }
        for(int i = N; i != 1; i= TT[i])
        {
            F[TT[i]][i] += Fmin;
            F[i][TT[i]] -= Fmin;
        }
        Flow += Fmin;
    }
    Print();
    return 0;
}