Cod sursa(job #3188584)

Utilizator AndreiBOTOBotocan Andrei AndreiBOTO Data 3 ianuarie 2024 13:42:09
Problema Felinare Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.11 kb
#include <bits/stdc++.h>

#pragma optimize GCC ("Ofast")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")

///#include <tryhardmode>
///#include <GODMODE::ON>

using namespace std;

ifstream fin ("felinare.in");
ofstream fout ("felinare.out");

const int NMAX=2e4+5;

vector<int>v[NMAX];

bool viz[NMAX];
int st[NMAX];
int dr[NMAX];

bool pairup(int p)
{
    if(viz[p])
        return 0;
    viz[p]=true;
    for(auto i:v[p])
    {
        if(!dr[i])
        {
            dr[i]=p;
            st[p]=i;
            return true;
        }
    }
    for(auto i:v[p])
    {
        if(pairup(dr[i]))
        {
            dr[i]=p;
            st[p]=i;
            return true;
        }
    }
    return false;
}

void mvc(int p)
{
    for(auto i:v[p])
    {
        if(!viz[i])
        {
            viz[i]=true;
            viz[st[i]]=false;
            mvc(st[i]);
        }
    }
}

int main()
{
    ios_base::sync_with_stdio(false);
    fin.tie(NULL);
    fout.tie(NULL);

    int n,m,i,j,kon=0;
    fin>>n>>m;
    for(i=1;i<=m;i++)
    {
        int x,y;
        fin>>x>>y;
        v[x].push_back(n+y);
        v[n+y].push_back(x);
    }
    bool ok=true;
    while(ok)
    {
        ok=false;
        for(i=1;i<=2*n;i++)
            viz[i]=false;
        for(i=1;i<=2*n;i++)
        {
            if(!st[i])
            {
                bool pairs=pairup(i);
                if(pairs==true)
                    ok=true;
                kon+=pairs;
            }
        }
    }
    fout<<2*n-kon/2<<"\n";
    for(i=1;i<=2*n;i++)
        viz[i]=false;
    for(i=1;i<=n;i++)
        if(dr[i])
            viz[i]=true;
    for(i=1;i<=n;i++)
        if(!dr[i])
            mvc(i);
    for(i=1;i<=n;i++)
    {
        if(viz[i]==true && viz[n+i]==true)
            fout<<0<<"\n";
        else if(viz[i]==true && viz[n+i]==false)
            fout<<2<<"\n";
        else if(viz[i]==false && viz[n+i]==false)
            fout<<3<<"\n";
        else
            fout<<1<<"\n";
    }
    fin.close();
    fout.close();
    return 0;
}