Cod sursa(job #2873087)

Utilizator alexmorosanuMorosanu Alexandru alexmorosanu Data 18 martie 2022 17:15:35
Problema Cuplaj maxim de cost minim Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.13 kb
#include <fstream>
#include <queue>
#include <vector>
#define INF 2000000001
using namespace std;
ifstream f("cmcm.in");
ofstream g("cmcm.out");
queue <int> q;
vector <int> v[1011];
bool inq[1011];
int ante[1011],dist[1011],cap[1011][1011],muc[1011][1011],cost[1011][1011],fol[1011][1011],d;
int bellman_ford()
{
    for(int i=0;i<=d;i++)
    {
        ante[i]=-1;
        dist[i]=INF;
        inq[i]=0;
    }
    q.push(0);
    inq[0]=1;
    dist[0]=0;
    while(q.empty()==false)
    {
        int k=q.front();
        q.pop();
        inq[k]=0;
        for(auto it=v[k].begin();it!=v[k].end();it++)
        {
            if(cap[k][*it]<=fol[k][*it] || dist[*it]<=dist[k]+cost[k][*it])
                continue;
            dist[*it]=dist[k]+cost[k][*it];
            ante[*it]=k;
            if(inq[*it]==0)
            {
                q.push(*it);
                inq[*it]=1;
            }
        }
    }
    if(dist[d]==INF)
        return 0;
    int M=INF;
    for(int i=d;i!=0;i=ante[i])
        M=min(M,cap[ante[i]][i]-fol[ante[i]][i]);
    for(int i=d;i!=0;i=ante[i])
    {
        fol[ante[i]][i]=fol[ante[i]][i]+M;
        fol[i][ante[i]]=fol[i][ante[i]]-M;
    }
    return M*dist[d];
}
int n,m,i,x,y,z,schimb=1,sol,nr,p;
int main()
{
    f>>n>>m>>p;
    d=n+m+1;
    for(i=1;i<=p;i++)
    {
        f>>x>>y>>z;
        y=y+n;
        v[x].push_back(y);
        v[y].push_back(x);
        cost[x][y]=z;
        cost[y][x]=-z;
        cap[x][y]=1;
        v[0].push_back(x);
        v[x].push_back(0);
        cap[0][x]=1;
        v[y].push_back(d);
        v[d].push_back(y);
        cap[y][d]=1;
        muc[x][y]=i;
    }
    while(schimb)
    {
        schimb=bellman_ford();
        sol=sol+schimb;
    }
    for(i=1;i<=n;i++)
        for(auto it=v[i].begin();it!=v[i].end();it++)
            if(*it!=0 && cap[i][*it] && fol[i][*it])
                nr++;
    g<<nr<<" "<<sol<<'\n';
    for(i=1;i<=n;i++)
        for(auto it=v[i].begin();it!=v[i].end();it++)
            if(*it!=0 && cap[i][*it]!=0 && fol[i][*it])
                g<<muc[i][*it]<<" ";
    return 0;
}