Cod sursa(job #2695520)

Utilizator mjmilan11Mujdar Milan mjmilan11 Data 13 ianuarie 2021 15:51:23
Problema Cuplaj maxim de cost minim Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.09 kb
#include <bits/stdc++.h>
using namespace std;
ifstream fin("cmcm.in");
ifstream fin2("cmcm.in");
ofstream fout("cmcm.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,nr,E;
bool ver[NMAX];
vector <int> v[NMAX];
int Bellman_Ford(){
    queue <int> q;
    for(int i=1;i<=n+m+2;i++){
        d[i]=INF;
        ver[i]=false;
    }
    q.push(S);
    ver[S]=true;
    d[S]=0;
    while(!q.empty()){
        int nod=q.front();
        ver[nod]=false;
        q.pop();
        if(nod==D) continue;
        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;
        nr=1;
        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 >> E;
    S=n+m+1;
    D=S+1;
    for(int i=1;i<=n;i++){
        v[S].push_back(i);
        c[S][i]=1;
        cost[S][i]=0;
    }
    for(int i=1;i<=m;i++){
        v[n+i].push_back(D);
        c[n+i][D]=1;
        cost[n+i][D]=0;
    }
    for(int i=1;i<=E;i++){
        fin >> x >> y >> z;
        y+=n;
        v[x].push_back(y);
        v[y].push_back(x);
        c[x][y]=1;
        cost[x][y]=z;
        cost[y][x]=-z;
    }
    int rasp=0,cnt=-1;
    nr=1;
    while(nr==1){
        nr=0;
        rasp+=Bellman_Ford();
        cnt++;
    }
    fout << cnt << ' ' << rasp << '\n';
    fin2 >> n >> m >> E;
    for(int i=1;i<=E;i++){
        fin2 >> x >> y >> z;
        if(f[x][y+n]>0) fout << i << ' ';
    }
    return 0;
}