Cod sursa(job #2707836)

Utilizator Uriesu_IuliusUriesu Iulius Uriesu_Iulius Data 17 februarie 2021 19:46:21
Problema 2SAT Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.73 kb
#include <bits/stdc++.h>

using namespace std;

ifstream fin("2sat.in");
ofstream fout("2sat.out");

int n, m, nrc;
int sol[100001];
vector<int> G[200001], GT[200001];
int ctc[200001];
bool ap[200001];
stack<int> stiva;

int non(int x)
{
    if(x <= n)
        return x + n;
    return x - n;
}

void Citire()
{
    int i, a, b;
    fin >> n >> m;
    for(i = 1; i <= m; i++)
    {
        fin >> a >> b;
        if(a < 0)
            a = non(-a);
        if(b < 0)
            b = non(-b);
        G[non(a)].push_back(b);
        G[non(b)].push_back(a);
        GT[b].push_back(non(a));
        GT[a].push_back(non(b));
    }
}

void initializare()
{
    int i;
    for(i = 1; i <= 2 * n; i++)
        ap[i] = 0;
}

void dfs(int k)
{
    ap[k] = 1;
    for(int x : G[k])
        if(!ap[x])
            dfs(x);
    stiva.push(k);
}

void dfsGT(int k)
{
    ctc[k] = nrc;
    for(int x : GT[k])
        if(ctc[x] == 0)
            dfsGT(x);
}

void Kosaraju()
{
    int i;
    for(i = 1; i <= 2 * n; i++)
        if(!ap[i])
            dfs(i);
    while(!stiva.empty())
    {
        int x = stiva.top();
        stiva.pop();
        if(ctc[x] == 0)
        {
            nrc++;
            dfsGT(x);
        }
    }
}

bool Exista_Solutie()
{
    int i;
    for(i = 1; i <= n; i++)
        if(ctc[i] == ctc[i + n])
            return false;
    return true;
}

void Solutie()
{
    int i;
    for(i = 1; i <= n; i++)
        if(ctc[i] < ctc[i + n])
            fout << 0 << " ";
        else
            fout << 1 << " ";
}

int main()
{
    Citire();
    Kosaraju();
    if(!Exista_Solutie())
        fout << -1;
    else
        Solutie();
    return 0;
}