Cod sursa(job #2246187)

Utilizator RaduXD1Nicolae Radu RaduXD1 Data 26 septembrie 2018 19:24:00
Problema Felinare Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.87 kb
#include <fstream>
#include <queue>
using namespace std;
ifstream fin("felinare.in");
ofstream fout("felinare.out");
int n,m,e,i,ok,a,b,u[8200],st[8200],dr[8200],sol,f1[8200],f2[8200];
vector <int> v[8200];

int solve(int nod)
{
    int i;int vecin;
    if(u[nod]==1)
        return 0;
    u[nod]=1;
    for(i=0;i<v[nod].size();i++)
    {
        vecin=v[nod][i];
        if(dr[vecin]==0)
        {
            st[nod]=vecin;
            dr[vecin]=nod;
            sol++;
            return 1;
        }
    }
    for(i=0;i<v[nod].size();i++)
    {
        vecin=v[nod][i];
        if(solve(dr[vecin]))
        {
            st[nod]=vecin;
            dr[vecin]=nod;
            return 1;
        }
    }
    return 0;
}

void punct(int nod)
{
    int i,vecin,drvecin;
    for(i=0;i<v[nod].size();i++)
    {
        vecin=v[nod][i];drvecin=dr[vecin];
        if(f2[vecin]==0)
        {
            f2[vecin]=1;
            f1[drvecin]=0;
            punct(drvecin);
        }
    }
}

int main()
{
    fin>>n>>m;
    for(i=1;i<=m;i++)
    {
        fin>>a>>b;
        v[a].push_back(b);
    }
    ok=1;
    while(ok==1)
    {
        ok=0;
        for(i=1;i<=n;i++)
            u[i]=0;
        for(i=1;i<=n;i++)
            if(st[i]==0)
                ok=max(ok,solve(i));
    }
    for(i=1;i<=n;i++)
        if(st[i]!=0)
            f1[i]=1;
    for(i=1;i<=n;i++)
        if(st[i]==0)
            punct(i);
    sol=2*n;
    for(i=1;i<=n;i++)
        sol-=f1[i]+f2[i];
    fout<<sol<<"\n";
    for(i=1;i<=n;i++)
        if(f1[i]==1&&f2[i]==1)
            fout<<"0\n";
        else
            if(f1[i]==1)
                fout<<"2\n";
            else
                if(f2[i]==1)
                    fout<<"1\n";
                else
                    fout<<"3\n";
    fin.close();
    fout.close();
    return 0;
}