Cod sursa(job #2961585)

Utilizator Anastasia7Anastasia Anastasia7 Data 6 ianuarie 2023 18:47:59
Problema Cuplaj maxim in graf bipartit Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.2 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <climits>

using namespace std;
ifstream fin("cuplaj.in");
ofstream fout("cuplaj.out");
vector<vector<int> > capacity,adj_list;
vector<int>parent;
int n,m,e,t,s;

bool BFS()
{
    vector<bool> viz(t+1);
    queue<int> coada;
    coada.push(s);
    viz[s]=true;
    parent[s]=-1;
    while(!coada.empty())
    {
        int u=coada.front();
        coada.pop();
        for(auto v:adj_list[u])
        {
            if(viz[v]==false && capacity[u][v]!=0)
            {
                parent[v]=u;
                if(v==t)
                    return true;
                viz[v]=true;
                coada.push(v);
            }
        }
    }
    return false;
}
int EdmondsKarp()
{
    long maxFlow=0;
    while(BFS()==true)
    {
        int u,v=t;
        int pathFlow=INT_MAX;
        while(v!=s)
        {
            u=parent[v];
            if(capacity[u][v]<pathFlow)
                pathFlow=capacity[u][v];
            v=parent[v];
        }
        v=t;
        while(v!=s)
        {
            u=parent[v];
            capacity[u][v]-=pathFlow;
            capacity[v][u]+=pathFlow;
            v=parent[v];
        }
        maxFlow+=pathFlow;
    }
    return maxFlow;
}

void afisare()
{
    fout<<EdmondsKarp()<<"\n";
    for(int i=1; i<=n; i++)
        for(int j=n+1;j<=n+m; j++)
        {
            if(capacity[j][i]==1)
                fout<<i<<" "<<j-n<<"\n";
        }
}

void citire()
{
    fin>>n>>m>>e;
    s=0;
    t=n+m+1;
    adj_list.resize(t+1);
    parent.resize(t+1);
    capacity.resize(t+1, vector<int>(t+1));
    for(int i=1;i<=e;i++)
    {
        int a,b;
        fin>>a>>b;
        adj_list[a].push_back(b+n);
        adj_list[b+n].push_back(a);
        capacity[a][b+n]=1;
    }
    for(int i=1;i<=n;++i)
    {
        adj_list[s].push_back(i);
        adj_list[i].push_back(s);
        capacity[s][i]=1;
    }
    for(int i=n+1;i<=n+m;++i)
    {
        adj_list[i].push_back(t);
        adj_list[t].push_back(i);
        capacity[i][t] = 1;
    }
    afisare();
}

int main()
{
    citire();
    return 0;
}