Cod sursa(job #2962704)

Utilizator simonatudoroiuTudoroiu Simona simonatudoroiu Data 8 ianuarie 2023 23:16:10
Problema Taramul Nicaieri Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 3.58 kb
//Link solutie:https://www.infoarena.ro/job_detail/2958717?action=view-source
//Explicatie: Pentru a rezolva problema folosim flux. Se va realiza un graf in care
//nodurile sunt nodul de start, nodul terminal, si cate un nod pentru fiecare oras,
//aceste noduri fiind dublate, fiecare dintre aceste noduri vor avea muchii cu toate
// celelalte noduri, excluzand pe el insusi din seria dublata. Apoi, gradele de iesire vor reprezenta capacitatea de la
//nodul de start la prima seria de noduri ce reprezinta orasele, iar gradele de intrare
//vor reprezenta capacitatea de la a doua serie de noduri ce reprezinta orasele, la nodul
//terminal. Pentru restul de muchii, atribuim capacitatea 1 si facem flux. Muchiile saturate
//dintre cele doua serii de noduri reprezentand orasele vor reprezenta si drumurile construite.
//Complexitate: O(n*n^2)
#include <bits/stdc++.h>
using namespace std;
ifstream fin("harta.in");
ofstream fout("harta.out");
int n, graf[201][201],a,b,c,rgraf[201][201];
//functie pentru sortat muchiile
bool functie_sort(pair<int,int>a, pair<int, int> b)
{
    return a.second<b.second;
}
//bfs pentru determinarea drumurilor de lungime minima de la sursa la nodul terminal
bool bfs(int rgraf[201][201], int tata[], int n)
{
    bool vizitat[2*n+2];
    for(int i=0;i<=2*n+1;i++)
        vizitat[i] = false;

    queue<int> q;
    q.push(0);
    vizitat[0] = true;
    tata[0] = -1;

    while(!q.empty())
    {
        int u = q.front();
        q.pop();

        for(int i = 0; i<=2*n+1;i++)
        {
            if(vizitat[i] == false && rgraf[u][i] > 0)
            {
                if(i == 2*n+1)
                {
                    tata[i] = u;
                    return true;
                }

                q.push(i);
                tata[i] = u;
                vizitat[i] = true;
            }
        }
    }

    return false;

}
int main() {
    fin>>n;
    //intitalizez graful
    for(int i=1;i<=n;i++)
    {
        fin>>a>>b;
        graf[0][i] = a;
        graf[n+i][2*n+1] = b;
    }
    //pun 1 pentru restul muchiilor
    for(int i=1;i<=n;i++)
    {
        for(int j=n+1;j<=2*n;j++)
        {
            if(j!=n+i)
            {
                graf[i][j] = 1;
            }
        }
    }
    //transfer totul in graful rezidual
    for(int i=0;i<=2*n+1;i++)
        for(int j=0;j<=2*n+1;j++)
        {
            rgraf[i][j] = graf[i][j];
        }

    int tata[2*n+2];
    //cat timp exista un drum de la sursa la nodul terminal
    while(bfs(rgraf, tata, n))
    {
        //revizuiesc fluxul pe drum
        int capacitate_drum = INT_MAX;

        for(int i=2*n+1; i!=0; i = tata[i])
        {
            int u = tata[i];
            capacitate_drum = min(capacitate_drum, rgraf[u][i]);
        }
        //actualizez graful rezidual
        for(int i=2*n+1; i!=0; i = tata[i])
        {
            int u = tata[i];
            rgraf[u][i] -= capacitate_drum;
            rgraf[i][u] += capacitate_drum;
        }
    }

    int nr_muchii=0;
    //determin muchiile saturate, pentru a determina cuplajul maxim
    vector<pair<int,int>> muchii;
    for(int i=n+1;i<=2*n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(rgraf[i][j] == 1)
            {
                nr_muchii++;
                muchii.push_back(make_pair(i-n, j));
            }
        }
    }
    sort(muchii.begin(),muchii.end(), functie_sort);
    //afisare
    fout<<nr_muchii<<endl;
    for(pair<int,int> i : muchii)
    {
        fout<<i.second<<" "<<i.first<<endl;
    }
    return 0;
}