Cod sursa(job #2789933)

Utilizator bogdanvladmihaiBogdan Vlad-Mihai bogdanvladmihai Data 28 octombrie 2021 10:32:09
Problema Andrei Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.75 kb
#include <bits/stdc++.h>

using namespace std;

ifstream in("andrei.in");
ofstream out("andrei.out");

const int maxN = 200000;

int n, m;

vector<int> g[maxN];

vector<int> r[maxN];

vector<int> st;

bitset<maxN> used, comp;

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

void dfs(int &u) {
  used[u] = 1;
  for (auto &v : g[u]) {
    if (used[v] == 0) {
      dfs(v);
    }
  }
  st.push_back(u);
}

void dfs2(int &u) {
  used[u] = comp[non(u)] = 0;
  comp[u] = 1;
  for (auto &v : r[u]) {
    if (used[v] == 1) {
      dfs2(v);
    }
  }
}

int main() {
  int u, v, z, nu, nv;
  in >> n >> m;
  for (int i = 0; i < m; i++) {
    in >> u >> v >> z;
    u--; v--;
    if (z == 1) {
      u = non(u), v = non(v);
      nu = non(u), nv = non(v);
      g[nv].push_back(u);
      r[u].push_back(nv);
      g[nu].push_back(v);
      r[v].push_back(nu);
    } else if (z == 0) {
      nu = non(u), nv = non(v);
      g[nv].push_back(u);
      r[u].push_back(nv);
      g[nu].push_back(v);
      r[v].push_back(nu);
    } else {
      u = non(u);
      nu = non(u), nv = non(v);
      g[nv].push_back(u);
      r[u].push_back(nv);
      g[nu].push_back(v);
      r[v].push_back(nu);
      u = non(u), v = non(v);
      nu = non(u), nv = non(v);
      g[nv].push_back(u);
      r[u].push_back(nv);
      g[nu].push_back(v);
      r[v].push_back(nu);
    }
  }
  for (int i = 0; i < 2 * n; i++) {
    if (used[i] == 0) {
      dfs(i);
    }
  }
  for (int i = 2 * n - 1; i >= 0; i--) {
    if (used[st[i]] == 1 && used[non(st[i])] == 1) {
      dfs2(st[i]);
    }
  }
  for (int i = 0; i < n; i++) {
    out << !comp[i] << " ";
  }
  in.close(); out.close();
  return 0;
}