Cod sursa(job #1818955)

Utilizator cella.florescuCella Florescu cella.florescu Data 29 noiembrie 2016 23:15:32
Problema Andrei Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.49 kb
#include <bits/stdc++.h>

using namespace std;

const int MAXN = 2e5;

vector <int> g[MAXN + 1], gt[MAXN + 1], srt;
int val[MAXN + 1], seen[MAXN + 1], n;

inline int non(int node) {
  if (node > n)
    return node - n;
  return node + n;
}

inline void add_edg(int x, int y) {
  g[x].push_back(y);
  gt[y].push_back(x);
}

void dfs(int node) {
  seen[node] = 1;
  for (auto it : g[node])
    if (seen[it] == 0)
      dfs(it);
  srt.push_back(node);
}

void rev_dfs(int node) {
  seen[node] = 0;
  val[non(node)] = 1;
  for (auto it : gt[node])
    if (seen[it])
      rev_dfs(it);
}

int main()
{
    int m, a, b, c;
    ifstream fin("andrei.in");
    fin >> n >> m;
    for (int i = 0; i < m; ++i) {
      fin >> a >> b >> c;
      switch (c) {
        case 0:
          add_edg(non(a), b);
          add_edg(non(b), a);
          break;
        case 1:
          add_edg(a, non(b));
          add_edg(b, non(a));
          break;
        case 2:
          add_edg(non(a), non(b));
          add_edg(b, a);
          add_edg(a, b);
          add_edg(non(b), non(a));
          break;
      }
    }
    fin.close();
    for (int i = 1; i <= 2 * n; ++i)
      if (seen[i] == 0)
        dfs(i);
    for (int i = 2 * n - 1; i >= 0; --i)
      if (seen[srt[i]] && seen[non(srt[i])])
        rev_dfs(srt[i]);
    ofstream fout("andrei.out");
    for (int i = 1; i <= n; ++i)
      fout << val[i] << " ";
    fout << '\n';
    fout.close();
    return 0;
}