Cod sursa(job #2774162)

Utilizator stefantagaTaga Stefan stefantaga Data 9 septembrie 2021 22:47:33
Problema Party Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.11 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>>x>>y>>tip;
        if (tip==0)
        {
            x=x;y=y;
        }
        else
        if (tip==1)
        {
            x=x;y=-y;
        }
        else
        if (tip==2)
        {
            x=-x;y=y;
        }
        else
        if (tip==3)
        {
            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=2*n;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;
        }
    }
    for (i=1;i<=n;i++)
    {
        val[i]=(grupa[i]>grupa[i+n]);
    }

    int numar=0;
    for (i=1;i<=n;i++)
    {
        if (val[i]==1)
        {
            numar++;
        }
        val[i+n]=1-val[i];
    }
    g<<numar<<'\n';
    for (i=1;i<=n;i++)
    {
        if (val[i]==1)
        {
            g<<i<<'\n';
        }
    }
    return 0;
}