Cod sursa(job #2047865)

Utilizator vladbatalanBatalan Vlad vladbatalan Data 25 octombrie 2017 14:54:07
Problema Flux maxim de cost minim Scor 50
Compilator cpp Status done
Runda Arhiva educationala Marime 2.86 kb
#include <bits/stdc++.h>
#define oo 100000000

using namespace std;

ifstream fin("fmcm.in");
ofstream fout("fmcm.out");

int n, m, s, d, flux, fmint, cost;
vector<int> v[1010];
int C[1010][1010], F[1010][1010], S[1010][1010];
int in_queue[1010], D[1010], TT[1010];
queue<int> coada;

void read()
{
    int x, y, c, z;
    fin >> n >> m >> s >> d;
    for(int i=1; i<=m; i++)
    {
        fin >> x >> y >> c >> z;
        C[x][y] += c;
        S[x][y] = -z;
        S[y][x] = z;
        v[x].push_back(y);
        v[y].push_back(x);
    }
}

//void afis(int A[][1010], int sm, char c)
//{
//    fout<<c<<":\n";
//    for(int i=1; i<=sm; i++)
//    {
//        for(int j=1; j<=sm; j++)
//            fout<<A[i][j]<<' ';
//        fout<<'\n';
//    }
//}

int bellman_ford()
{
    int top;
    for(int i=1; i<=n; i++){
        D[i] = oo;
        TT[i] = -1;
    }
    while(!coada.empty())
        coada.pop();
    coada.push(s);
    D[s] = 0;
    in_queue[s] = 1;
//    fout<<"in coada: ";
    while(!coada.empty())
    {
//        cout<<"#";
        top = coada.front();
//        fout<<top<<'\n';
        coada.pop();
        in_queue[top] = 0;
        for(auto vec:v[top])
        {
            if(D[vec] > D[top] + S[vec][top] && F[top][vec] != C[top][vec] && vec!=s)
            {
//                fout<<"\timbunatatim "<<top<<"-"<<vec<<'\n';
                D[vec] = D[top] + S[vec][top];
                TT[vec] = top;
                if(!in_queue[vec] && vec!=d)
                {
                    coada.push(vec);
                    in_queue[vec] = 1;
                }
//                fout<<"dimensiuni: ";
//                for(int i=1; i<=n; i++)
//                    fout<<D[i]<<' ';
//                fout<<'\n';
            }
        }
    }
//    fout<<"out coada\n";
    return D[d];
}

int main()
{
    read();
    while(bellman_ford() != oo)
    {
//        for(int i=0; i<v[d].size(); i++)
//        {
//            if(D[v[d][i]] == oo || C[v[d][i]][d] - F[v[d][i]][d] == 0) continue;
//            TT[d]=v[d][i];

//            fout<<"TATI: ";
//            for(int i=1; i<=n; i++)
//                fout<<TT[i]<<' ';
//            fout<<'\n';

//            fout<<"dimensiuni: ";
//            for(int i=1; i<=n; i++)
//                fout<<D[i]<<' ';
//            fout<<'\n';

            fmint = oo;
            for(int nod = d; nod != s; nod = TT[nod])
                fmint = min(fmint, C[TT[nod]][nod] - F[TT[nod]][nod]);
            if(fmint == 0) continue;

            for(int nod = d; nod != s; nod = TT[nod])
            {
                cost += fmint*S[TT[nod]][nod];
                F[TT[nod]][nod] += fmint;
                F[nod][TT[nod]] -= fmint;
            }
//            afis(F, n, 'F');
//        }
    }
    cost = -cost;
    fout<<cost;
    return 0;
}