Cod sursa(job #2774155)

Utilizator stefantagaTaga Stefan stefantaga Data 9 septembrie 2021 22:27:28
Problema Party Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 3.74 kb
#include <bits/stdc++.h>

using namespace std;
ifstream f("party.in");
ofstream g("party.out");
int n,val[400005],viz[400005],sal[400005],q,nr;
vector <int> v[400005],inv[400005],comp[400005];
int nod(int x)
{
    if (x<0)
    {
        return n+(-x);
    }
    return x;
}
void adauga(int x,int y)
{
    x=nod(x),y=nod(y);
    v[x].push_back(y);
    inv[y].push_back(x);
}
void dfs(int x)
{
    viz[x]=1;
    for (int i=0;i<v[x].size();i++)
    {
        int nod=v[x][i];
        if (viz[nod]==0)
        {
            dfs(nod);
        }
    }
    sal[++q]=x;
}
void dfs2(int x)
{
    viz[x]=1;
    comp[nr].push_back(x);
    for (int i=0;i<inv[x].size();i++)
    {
        int nod=inv[x][i];
        if (viz[nod]==0)
        {
            dfs2(nod);
        }
    }
}
int m,i,j,x,y,tip,grupa[400005],nrintr[400005];
pair <int,int> ceau[400005];
int main()
{
    f>>n>>m;
    for (i=1;i<=m;i++)
    {
        f>>tip>>x>>y;
        if (tip==0)
        {
            x=x;y=y;
        }
        else
        if (tip==1)
        {
            x=x;y=-y;
        }
        else
        if (tip==2)
        {
            x=-x;y=y;
        }
        else
        {
            x=-x;y=-y;
        }
        ceau[i]={x,y};
        adauga(-x,y);
        adauga(-y,x);
    }
    for (i=1;i<=2*n;i++)
    {
        if (viz[i]==0)
        {
            dfs(i);
        }
    }
    nr=0;
    memset(viz,0,sizeof(viz));
    for (i=q;i>=1;i--)
    {
        if (viz[sal[i]]==0)
        {
            nr++;
            dfs2(sal[i]);
        }
    }
    for (i=1;i<=nr;i++)
    {
        for (j=0;j<comp[i].size();j++)
        {
            grupa[comp[i][j]]=i;
        }
    }
    memset(val,-1,sizeof(val));
    for (i=1;i<=2*n;i++)
    {
        for (j=0;j<v[i].size();j++)
        {
            int nod=v[i][j];
            if (grupa[i]!=grupa[nod])
            {
                nrintr[grupa[nod]]++;
            }
        }
    }
    for (i=1;i<=n;i++)
    {
        if (grupa[i]==grupa[i+n])
        {
            g<<"-1";
            return 0;
        }
    }
    queue <int> coada;
    for (i=1;i<=nr;i++)
    {
        if (nrintr[i]==0)
        {
            coada.push(i);
        }
    }
    while (!coada.empty())
    {
        int acum=coada.front();
        coada.pop();
        for (int i=0;i<comp[acum].size();i++)
        {
            int nod=comp[acum][i];
            if (nod<=n)
            {
                if (val[nod]==-1)
                {
                    val[nod]=0;
                }
            }
            else
            {
                if (val[nod-n]==-1)
                {
                    val[nod-n]=1;
                }
            }
        }
        for (int i=0;i<comp[acum].size();i++)
        {
            int nod=comp[acum][i];
            for (int j=0;j<v[nod].size();j++)
            {
                if (grupa[nod]!=grupa[v[nod][j]])
                {
                    nrintr[grupa[v[nod][j]]]--;
                    if (nrintr[grupa[v[nod][j]]]==0)
                    {
                        coada.push(grupa[v[nod][j]]);
                    }
                }
            }
        }
    }
    int numar=0;
    for (i=1;i<=n;i++)
    {
        if (val[i]==1)
        {
            numar++;
        }
        val[i+n]=1-val[i];
    }
    for (i=1;i<=m;i++)
    {
        x=nod(ceau[i].first);
        y=nod(ceau[i].second);
        if ((val[x]|val[y])==0)
        {
            g<<"NU"<<" "<<x<<" "<<y;
            return 0;
        }
    }
    g<<numar<<'\n';
    for (i=1;i<=n;i++)
    {
        if (val[i]==1)
        {
            g<<i<<'\n';
        }
    }
    return 0;
}