Cod sursa(job #2958717)

Utilizator simonatudoroiuTudoroiu Simona simonatudoroiu Data 27 decembrie 2022 22:17:55
Problema Taramul Nicaieri Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.38 kb
#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];
bool functie_sort(pair<int,int>a, pair<int, int> b)
{
    return a.second<b.second;
}
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;
    for(int i=1;i<=n;i++)
    {
        fin>>a>>b;
        graf[0][i] = a;
        graf[n+i][2*n+1] = b;
    }

    for(int i=1;i<=n;i++)
    {
        for(int j=n+1;j<=2*n;j++)
        {
            if(j!=n+i)
            {
                graf[i][j] = 1;
            }
        }
    }

    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];

    while(bfs(rgraf, tata, n))
    {
        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]);
        }

        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;
    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);
    fout<<nr_muchii<<endl;
    for(pair<int,int> i : muchii)
    {
        fout<<i.second<<" "<<i.first<<endl;
    }
//    for(int i=0;i<=2*n+1;i++)
//    {
//        for(int j=0;j<=2*n+1;j++)
//        {
//            cout<<rgraf[i][j]<<" ";
//        }
//        cout<<endl;
//    }
    return 0;
}