Cod sursa(job #2628769)

Utilizator loraclorac lorac lorac Data 17 iunie 2020 14:04:38
Problema Traseu Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.85 kb
#include <fstream>
#include <vector>
#include <queue>
using namespace std;
ifstream cin("traseu.in");
ofstream cout("traseu.out");
const int inf=1e9+7;
vector<pair<int,int> > vec[70];
int out[70],in[70];
int c[70][70];
int pred[70];
int dist[70];
queue<int> q;
bool inq[70];
int main()
{
    int n,m,x,y,z;
    cin>>n>>m;
    for(int i=1;i<=m;++i)
    {
        cin>>x>>y>>z;
        vec[x].push_back({y,z});
        vec[y].push_back({x,-z});
        c[x][y]=inf;
        ++out[x];
        ++in[y];
    }
    for(int i=1;i<=n;++i)
    if(out[i]>in[i])
    {
        vec[0].push_back({i,0});
        vec[i].push_back({0,0});
        c[0][i]=out[i]-in[i];
    }
    else if(in[i]>out[i])
    {
        vec[i].push_back({n+1,0});
        vec[n+1].push_back({i,0});
        c[i][n+1]=in[i]-out[i];
    }
    int cost=0,flow=0,ads;
    do
    {
        for(int i=0;i<=n+1;++i)
        {
            pred[i]=-1;
            dist[i]=inf;
        }
        dist[0]=0;
        inq[0]=true;
        q.push(0);
        while(!q.empty())
        {
            int x=q.front(); q.pop();
            inq[x]=false;
            for(auto y:vec[x])
            if(c[x][y.first]>0 and dist[y.first]>dist[x]+y.second)
            {
                dist[y.first]=dist[x]+y.second;
                pred[y.first]=x;
                if(!inq[y.first]) inq[y.first]=true,q.push(y.first);
            }
        }
        if(pred[n+1]>=0)
        {
            ads=inf;
            for(int k=n+1;pred[k]>=0;k=pred[k])
                ads=min(ads,c[pred[k]][k]);
            for(int k=n+1;pred[k]>=0;k=pred[k])
            {
                c[pred[k]][k]-=ads;
                c[k][pred[k]]+=ads;
            }
            cost+=ads*dist[n+1];
            flow+=ads;
        }
    }while(pred[n+1]>=0);
    cout<<cost<<'\n';
    return 0;
}