Cod sursa(job #2415234)

Utilizator stefdascalescuStefan Dascalescu stefdascalescu Data 25 aprilie 2019 17:20:43
Problema Felinare Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.7 kb
#include<bits/stdc++.h>
using namespace std;
ifstream f("felinare.in");
ofstream g("felinare.out");
int n, m;
vector<int>v[8200];
int stanga[8200], dreapta[8200];
bool viz[8200];
bool scos[8200], scosb[8200];
bool dfs(int nod)
{
    if(viz[nod])
        return 0;
    viz[nod] = 1;
    for(int i = 0; i < v[nod].size(); i++)
    {
        int vecin = v[nod][i];
        if(!dreapta[vecin])
        {
            stanga[nod] = vecin;
            dreapta[vecin] = nod;
            return 1;
        }
    }
    for(int i = 0; i < v[nod].size(); i++)
    {
        int vecin = v[nod][i];
        if(dfs(dreapta[vecin]))
        {
            stanga[nod] = vecin;
            dreapta[vecin] = nod;
            return 1;
        }
    }
    return 0;
}
void dfs2(int nod)
{
    if(viz[nod])
        return;
    viz[nod] = 1;
    for(int i = 0; i < v[nod].size(); ++i)
    {
        int vecin = v[nod][i];
        if(stanga[nod] != vecin && !scosb[vecin])
        {
            scosb[vecin] = 1;
            dfs2(dreapta[vecin]);
        }
    }
}
int main()
{
    f >> n >> m;
    for(int i = 1; i <= m; ++i)
    {
        int a, b;
        f >> a >> b;
        v[a].push_back(b);
    }
    bool ok = 1;
    while(ok)
    {
        ok = 0;
        memset(viz, 0, sizeof(viz));
        for(int i = 1; i <= n; ++i)
            if(!stanga[i] && dfs(i))
                ok = 1;
    }
    int ans = 2 * n;
    memset(viz, 0, sizeof(viz));
    for(int i = 1; i <= n; ++i)
        if(stanga[i])
            --ans;
        else
            dfs2(i);
    g << ans << '\n';
    for(int i = 1; i <= n; ++i)
        g << viz[i] + 2 * (1 - scosb[i]) << '\n';
    return 0;
}