Cod sursa(job #2694551)

Utilizator VladMxPMihaila Vlad VladMxP Data 9 ianuarie 2021 17:50:55
Problema Felinare Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.38 kb
#include <bits/stdc++.h>

using namespace std;
ifstream fin("felinare.in");
ofstream fout("felinare.out");
int st[10000],dr[10000],viz[10000],n,m,ok,x,y,nr;
bool apst[10000],apdr[10000];
vector <int> v[10000];

int pairup(int i)
{
    if(viz[i]) return 0;
    viz[i]=1;
    for(int j=0; j<v[i].size(); j++)
    {
        int vec=v[i][j];
        if(!dr[vec] || pairup(dr[vec]))
        {
            st[i]=vec;
            dr[vec]=i;
            apst[i]=1;
            return 1;
        }
    }
    return 0;
}

void dfs(int i)
{
    for(int j=0;j<v[i].size();j++)
    {
        int vec=v[i][j];
        if(!apdr[vec])
        {
            apdr[vec]=1;
            apst[dr[vec]]=0;
            dfs(dr[vec]);
        }
    }
}
int main()
{
    int i;
    fin>>n>>m;
    for(i=1; i<=m; i++)
    {
        fin>>x>>y;
        v[x].push_back(y);
    }
    ok=1;
    while(ok)
    {
        memset(viz,0,sizeof(viz));
        ok=0;
        for(i=1; i<=n; i++)
            if(!st[i] && pairup(i)) ok=1;
    }
    for(i=1;i<=n;i++)
      if(st[i]) nr++;
    fout<<2*n-nr<<'\n';
    for(i=1;i<=n;i++)
        if(!apst[i]) dfs(i);
    for(i=1;i<=n;i++)
    {
        if(!apst[i] && !apdr[i]) fout<<3<<'\n';
        else if(apst[i] && !apdr[i]) fout<<2<<'\n';
        else if(!apst[i] && apdr[i]) fout<<1<<'\n';
        else fout<<0<<'\n';
    }
}