Cod sursa(job #2545545)

Utilizator iulianarsenoiuArsenoiu Iulian iulianarsenoiu Data 13 februarie 2020 11:34:10
Problema Felinare Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.78 kb
#include <fstream>
#include <cstring>
#include <vector>

using namespace std;
ifstream f("felinare.in");
ofstream g("felinare.out");
vector <int> G[10005];
int r[10005],l[10005];
bool sel[10005],st[10005],dr[10005];
int n,m;
bool cupleaza(int nod)
{
    sel[nod]=true;
    for(auto it : G[nod])
    {
        if(!r[it])
        {
            r[it]=nod;
            l[nod]=it;
            return true;
        }
    }
    for(auto it : G[nod])
    {
        if(!sel[r[it]] && cupleaza(r[it]))
        {
            r[it]=nod;
            l[nod]=it;
            return true;
        }
    }
    return false;
}
void dfs(int x)
{
    for(auto it : G[x])
    {
        if(!dr[it])
        {
            dr[it]=true;
            st[r[it]]=false;
            dfs(r[it]);
        }
    }
}
int main()
{
    f>>n>>m;
    for(int i=1;i<=m;i++)
    {
        int x,y;
        f>>x>>y;
        G[x].push_back(y);
    }
    bool done=false;
    while(!done)
    {
        done=true;
        memset(sel,false,sizeof(sel));
        for(int i=1;i<=n;i++)
        {
            if(!sel[i] && !l[i] && cupleaza(i))
            {
                done=false;
            }
        }
    }
    for(int i=1;i<=n;i++)
        if(l[i])
        st[i]=true;
    for(int i=1;i<=n;i++)
        if(!l[i])
        dfs(i);
    int nr=0;
    for(int i=1;i<=n;i++)
    {
        if(!st[i])
            ++nr;
        if(!dr[i])
            nr++;
    }
    g<<nr<<'\n';
    for(int i=1;i<=n;i++)
    {
        if(!st[i] && !dr[i])
        {
            g<<3<<'\n';
        }
        else if(!st[i])
        {
            g<<1<<'\n';
        }
        else if(!dr[i])
        {
            g<<2<<'\n';
        }
        else g<<0<<'\n';
    }
    return 0;
}