Cod sursa(job #2020775)

Utilizator BogdanisarBurcea Bogdan Madalin Bogdanisar Data 11 septembrie 2017 17:49:44
Problema 2SAT Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 3.6 kb
#include <iostream>
#include <fstream>
#include <stack>
#include <vector>
#include <deque>
#include <ctime>
#include <queue>
#include <cmath>

#define ll long long
#define ull unsigned long long
#define pb push_back
#define mp make_pair
using zint = int;
using namespace std;
const int inf = 3e4 + 5;
const int NMax = 1e5 + 5;
const int MMax = 2e5 + 5;
ifstream in("2sat.in");
ofstream out("2sat.out");

zint N,M,nrComp;
zint compOf[2*NMax],ans[NMax];
bool visStack[2*NMax],inComp[2*NMax];
vector<zint> v[2*NMax],rev[2*NMax],comp[2*NMax];
stack<zint> st;
// compOf[i] = indexul componentei tare conexe care contine nodul i;
// ans[i] = 0 || 1; valoarea finala a literalului i;
// visStack[i] - vector de vizite pentru prima parte a algoritmului lui Kosaraju;
// inComp[i] - vector de vizite pentru a doua parte a algoritmului lui Kosaraju;
//           = true daca s-a gasit componenta tare conexa in care se afla nodul i;
// v[i] - lista de adiacenta pentru nodul i in graful relatiilor de implicatie;
// rev[i] - lista de adiacenta pentru nodul i in opusul grafului relatiilor de implicatie;
// comp[i] - nodurile care se afla in componenta i;

void getStack(zint);
void getComp(zint);
// functii pentru algoritmul lui Kosaraju de determinare a componentelor tare conexe;
void update(zint,const zint);
// update(c,val) pune valoarea val in toti literalii din componenta c;

#define inv(x) ((x > N) ? x-N : x+N)
// nodurile negative se numesc de la N in sus;
// (e.g.: inv1 == N+1)
// invx - indexul negarii lui x;
int main() {
    in>>N>>M;

    for (int i=1;i <= N;++i) {
        ans[i] = -1;
    }

    for (int i=1;i <= M;++i) {
        zint x,y;
        in>>x>>y;

        x = (x < 0) ? -x + N : x;
        y = (y < 0) ? -y + N : y;

        v[inv(x)].pb(y);
        rev[y].pb(inv(x));
        v[inv(y)].pb(x);
        rev[x].pb(inv(y));
    }

    for (zint i = 1;i <= 2*N;++i) {
        if (visStack[i]) {
            continue;
        }

        getStack(i);
    }

    while (st.size()) {
        zint node = st.top(); st.pop();
        if (!inComp[node]) {
            ++nrComp;
            getComp(node);
        }
    }

    bool sol = true;
    for (zint i=1;i <= N;++i) {
        if (compOf[i] == compOf[i+N]) { // nu exista solutie
            sol = false;
            break;
        }
    }

    if (sol) {
        // algoritmul lui Kosaraju return-eaza componentele in ordinea de care avem nevoie;
        // asa ca doar se parcurg componentele si se updateaza elementele care nu au inca o valoare;
        for (zint c=1;c <= nrComp;++c) {
            update(c,0);
        }

        for (zint i=1;i <= N;++i) {
            out<<ans[i]<<' ';
        }
    }
    else {
        out<<"-1\n";
    }

    in.close();out.close();
    return 0;
}

void getStack(zint node) {
    visStack[node] = true;

    for (zint nxt : v[node]) {
        if (visStack[nxt]) {
            continue;
        }

        getStack(nxt);
    }

    st.push(node);
}

void getComp(zint node) {
    comp[nrComp].pb(node);
    compOf[node] = nrComp;
    inComp[node] = true;

    for (zint nxt : rev[node]) {
        if (inComp[nxt]) {
            continue;
        }

        getComp(nxt);
    }
}

void update(zint c,const zint val) {
    zint x = comp[c][0];
    x = (x > N) ? x-N : x;
    if (ans[x] != -1) { // s-a facut deja update la componenta c;
        return;
    }

    for (auto node : comp[c]) {
        zint temp = val;
        if (node > N) {
            node -= N;
            temp ^= 1;
        }
        ans[node] = temp;
    }
}