Cod sursa(job #1511869)

Utilizator fluture.godlikeGafton Mihnea Alexandru fluture.godlike Data 27 octombrie 2015 11:28:11
Problema Andrei Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.84 kb
#include <cstdio>
#include <vector>
#include <stack>

#define NMAX 200007
#define neg(a) (a^1)

using namespace std;
int n, m, x, y, p, viz[NMAX], sol[NMAX];
vector <int> g[NMAX], gt[NMAX];
stack <int> order;
void convert(int &a)
{
    if(a < 0) a = (((0-a)<<1)|1);
    else a = (a<<1);
}
void add(const int &a, const int &b)
{
    g[a].push_back(b);
    gt[b].push_back(a);
}
void create(const int &a, const int &b, const int &t)
{
    if(t == 0)
    {
        add(neg(a), b);
        add(neg(y), x);
    }
    if(t == 1)
    {
        add(x, neg(y));
        add(y, neg(x));
    }
    if(t == 2)
    {
        add(x, y);
        add(y, x);
        add(neg(x), neg(y));
        add(neg(y), neg(x));
    }
}
void dfsort(const int &nod)
{
    viz[nod] = 1;
    for(int i = g[nod].size()-1; i+1; --i)
    {
        if(!viz[g[nod][i]]) dfsort(g[nod][i]);
    }
    order.push(nod);
}
void dfsat(const int &nod)
{
    viz[nod] = 0;
    sol[neg(nod)] = 1;
    for(int i = gt[nod].size()-1; i+1; --i)
    {
        if(viz[gt[nod][i]]) dfsat(gt[nod][i]);
    }
}
void sortare()
{
    int tmp;
    for(int i = 1; i<= n; ++i)
    {
        tmp = (i<<1);
        if(!viz[tmp]) dfsort(tmp);
        tmp = (tmp|1);
        if(!viz[tmp]) dfsort(tmp);
    }
}
void sat()
{
    int tmp;
    while(!order.empty())
    {
        tmp = order.top();
        order.pop();
        if(viz[tmp] && viz[neg(tmp)]) dfsat(tmp);
    }
}

int main()
{
    freopen("andrei.in", "r", stdin);
    freopen("andrei.out", "w", stdout);
    scanf("%d %d", &n, &m);
    for(int i = 1; i<= m; ++i)
    {
        scanf("%d %d %d", &x, &y, &p);
        convert(x);convert(y);
        create(x, y, p);
    }
    sortare();
    sat();
    for(int i = 1; i<= n; ++i) printf("%d ", sol[(i<<1)]);
    printf("\n");
    return 0;
}