Cod sursa(job #2639814)

Utilizator cyg_vladioanBirsan Vlad cyg_vladioan Data 3 august 2020 23:37:41
Problema Felinare Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.88 kb
#include <cstdio>
#include <vector>
#include <cstring>
using namespace std;
const int NMAX = 8191;
vector <int> G[NMAX + 5];
int l[NMAX + 5] , r[NMAX + 5] , viz[NMAX + 5] , st[NMAX + 5] , dr[NMAX + 5];
int path(int u)
{
    int v , j;
    if(viz[u])
        return 0;
    viz[u] = 1;
    for(j = 0 ; j < G[u].size() ; j ++)
    {
        v = G[u][j];
        if(!r[v])
        {
            l[u] = v;
            r[v] = u;
            return 1;
        }
    }
    for(j = 0 ; j < G[u].size() ; j ++)
    {
        v = G[u][j];
        if(path(r[v]))
        {
            l[u] = v;
            r[v] = u;
            return 1;
        }
    }
    return 0;
}
void dfs(int u)
{
    int v , j;
    viz[u] = 1;
    st[u] = 0;
    for(j = 0 ; j < G[u].size() ; j ++)
    {
        v = G[u][j];
        if(!viz[r[v]])
        {
            dr[v] = 1;
            dfs(r[v]);
        }
    }
}
int main()
{
    freopen("felinare.in" , "r" , stdin);
    freopen("felinare.out" , "w" , stdout);
    int n , m , i , cuplaj , x , y , gasit;
    scanf("%d%d" , &n , &m);
    for(i = 1 ; i <= m ; i ++)
    {
        scanf("%d%d" , &x , &y);
        G[x].push_back(y);
    }
    cuplaj = 0;
    do
    {
        gasit = 0;
        memset(viz , 0 , sizeof(viz));
        for(i = 1 ; i <= n ; i ++)
            if(!l[i] && path(i))
            {
                cuplaj ++;
                gasit = 1;
            }
    }
    while(gasit == 1);
    memset(viz , 0 , sizeof(viz));
    for(i = 1 ; i <= n ; i ++)
        if(l[i])
            st[i] = 1;
    for(i = 1 ; i <= n ; i ++)
        if(!l[i])
            dfs(i);
    printf("%d\n" , 2 * n - cuplaj);
    for(i = 1 ; i <= n ; i ++)
    {
        int nr = 0;
        if(!st[i])
            nr += 1;
        if(!dr[i])
            nr += 2;
        printf("%d\n" , nr);
    }
    return 0;
}