Cod sursa(job #2961899)

Utilizator biancar28Radulescu Alexia-Bianca biancar28 Data 7 ianuarie 2023 12:40:35
Problema Cuplaj maxim in graf bipartit Scor 90
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.97 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <climits>
#include <bits/stdc++.h>
using namespace std;

ifstream f("cuplaj.in");
ofstream g("cuplaj.out");

int l,r,m,n,x,y,c,s,d;
vector<int>viz;
vector<int>t;//vector de muchii care duc la nodurile respective
vector<vector<int>>indici; //vector de indici de muchii;
// indici[i] = vector de muchii care ies din i

struct muchie{
    int x,y,c,poz;
};

vector<muchie>M;

int BFS(){
    t.clear();
    t.resize(n);
    fill(viz.begin(), viz.end(), 0);

    queue<int>q;
    viz[s]=1;
    q.push(s);

    while(!q.empty()){
        int vf = q.front();
        q.pop();
        for(auto aux:indici[vf]){
            muchie mc = M[aux];
            if(!viz[mc.y] && mc.c>0){
                q.push(mc.y);
                viz[mc.y] = 1;
                t[mc.y] = aux;
                if(mc.y==d){
                    return 1;
                }
            }
        }
    }
   return 0;
}


int main(){
    f>>l>>r>>m;
    n=l+r+2;
    s=0;
    d = n-1;

    viz.resize(n);
    t.resize(n);
    indici.resize(n);

    int dim = 0;
    for(int i=1;i<=m;i++){
        f>>x>>y;
        dim +=2;
        M.push_back({x,y+l,1,dim-1});
        M.push_back({y+l,x,0,dim-2});
        indici[x].push_back(dim-2);
        indici[y+l].push_back(dim-1);
    }

    //adaug muchii de la sursa spre fiecare nod din stanga
    for(int i=1;i<=l;i++){
        dim+=2;
        M.push_back({s,i,1,dim-1});
        M.push_back({i,s,0,dim-2});
        indici[s].push_back(dim-2);
        indici[i].push_back(dim-1);
    }

    for(int i = 1+l; i<=r+l; i++){
        dim+=2;
        M.push_back({i,d,1,dim-1});
        M.push_back({d,i,0,dim-2});
        indici[i].push_back(dim-2);
        indici[d].push_back(dim-1);
    }

    for(int i = 0; i<M.size();i++){
        cout<<i<<": "<<M[i].x<<" "<<M[i].y<<" "<<M[i].c<<" "<<M[i].poz<<"\n";
    }

    cout<<"\n\n";
    for(int i=0;i<=d;i++){
        cout<<i<<": ";
        for(int j=0;j<indici[i].size();j++){
            cout<<indici[i][j]<<" ";
        }
        cout<<"\n";
    }


    c=0;
    while(BFS()){
        for(auto aux:indici[d]){
            if(viz[M[aux].y] && M[M[aux].poz].c!=0){
                int minim = INT_MAX;
                muchie mc = M[aux];
                t[d] = mc.poz;
                int nod = d;

                while(nod != s){
                    minim = min(minim, M[t[nod]].c);
                    nod = M[t[nod]].x;
                }
                c +=minim;
                nod = d;
                while(nod!=s){
                    M[t[nod]].c -= minim;
                    M[M[t[nod]].poz].c +=minim;
                    nod = M[t[nod]].x;
                }
            }
        }
    }
    g<<c<<"\n";

    //nodurile cuplate fac parte din arcele care nu au extremitatile in sursa
    // sau destinatie si au capacitatea 0

    for(auto aux: M){
        if(aux.c==0 && aux.x!=s && aux.y!=d && aux.x<aux.y){
            g<<aux.x << " " << aux.y-l<<"\n";
        }
    }

    return 0;
}