Cod sursa(job #2001447)

Utilizator refugiatBoni Daniel Stefan refugiat Data 16 iulie 2017 17:53:11
Problema Andrei Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.55 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <stack>
#include <bitset>
#define NMAX 200005
using namespace std;
ifstream si("andrei.in");
ofstream so("andrei.out");
vector<int> v[NMAX],t[NMAX];
stack<int> s;
bitset<NMAX> viz;
int n,is[NMAX],sol[NMAX];
int niv[NMAX],low[NMAX];
int neg(int x)
{
    if(x<=n)
        return x+n;
    return x-n;
}
void add(int x,int y)
{
    v[neg(x)].push_back(y);
    t[y].push_back(neg(x));
    v[neg(y)].push_back(x);
    t[x].push_back(neg(y));
}

void dfs(int x)
{
    viz[x]=1;
    int y;
    for(int i=0;i<v[x].size();i++)
    {
        y=v[x][i];
        if(!viz[y])
            dfs(y);
    }
    s.push(x);
}
void dft(int x)
{
    if(sol[x])
    {
        sol[0]=-1;
        return;
    }
    int i,y;
    sol[neg(x)]=1;
    viz[x]=0;
    for(i=0;i<t[x].size();i++)
    {
        y=t[x][i];
        if(viz[y]==1)
            dft(y);
    }
}
int main()
{
    int m;
    si>>n>>m;
    for(int i=1;i<=m;++i)
    {
        int a,b,c;
        si>>a>>b>>c;
        if(c==0)
        {
            add(a,b);
        }
        if(c==1)
            add(neg(a),neg(b));
        else
        {
            add(neg(a),b);
            add(a,neg(b));
        }
    }
    for(int i=1;i<=n*2;++i)
        if(!viz[i])
            dfs(i);
    while(!s.empty())
    {
        int x=s.top();
        s.pop();
        if(viz[x]&&viz[neg(x)])
        {
            dft(x);
        }
    }
    for(int i=1;i<=n;++i)
        so<<sol[i]<<' ';
    return 0;
}