Cod sursa(job #2694725)

Utilizator mjmilan11Mujdar Milan mjmilan11 Data 10 ianuarie 2021 16:57:58
Problema Flux maxim de cost minim Scor 20
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.57 kb
#include <bits/stdc++.h>
using namespace std;
ifstream fin("fmcm.in");
ofstream fout("fmcm.out");
const int INF = (1<<29);
const int NMAX = 355;
int f[NMAX][NMAX],c[NMAX][NMAX],cost[NMAX][NMAX],d[NMAX],tata[NMAX];
int S,D,n,m,x,y,z,cc;
bool ver[NMAX];
vector <int> v[NMAX];
int Bellman_Ford(){
    queue <int> q;
    for(int i=1;i<=n;i++){
        d[i]=INF;
        ver[i]=false;
    }
    q.push(S);
    ver[S]=true;
    d[S]=0;
    while(!q.empty()){
        int nod=q.front();
        q.pop();
        for(int i=0;i<v[nod].size();i++){
            int vecin=v[nod][i];
            if(c[nod][vecin]>f[nod][vecin] and d[vecin]>d[nod]+cost[nod][vecin]){
                d[vecin]=d[nod]+cost[nod][vecin];
                if(ver[vecin]==false){
                    ver[vecin]=true;
                    q.push(vecin);
                }
                tata[vecin]=nod;
            }
        }
    }
    if(d[D]!=INF){
        int minim=INF;
        for(int i=D;i!=S;i=tata[i]) minim=min(minim,c[tata[i]][i]-f[tata[i]][i]);
        for(int i=D;i!=S;i=tata[i]) {
            f[tata[i]][i]+=minim;
            f[i][tata[i]]-=minim;
        }
        return minim*d[D];
    }
    return 0;
}

int main()
{
    fin >> n >> m >> S >> D;
    for(int i=1;i<=m;i++){
        fin >> x >> y >> cc >> z;
        v[x].push_back(y);
        v[y].push_back(x);
        c[x][y]=cc;
        cost[x][y]=z;
        cost[y][x]=-z;
    }
    int rasp=0,nr;
    do{
        nr=Bellman_Ford();
        rasp+=nr;
    } while(nr!=0);
    fout << rasp;
    return 0;
}