Submission #1101514

---

Source Code Expand

#include <algorithm>
#include <cassert>
#include <cfloat>
#include <climits>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <deque>
#include <iomanip>
#include <iostream>
#include <limits>
#include <map>
#include <queue>
#include <set>
#include <sstream>
#include <stack>
#include <string>
#include <tuple>
#include <vector>

#define FOR(i,k,n) for (int (i)=(k); (i)<(n); ++(i))
#define rep(i,n) FOR(i,0,n)
#define pb push_back
#define all(v) begin(v), end(v)
#define debug(x) cerr<< #x <<": "<<x<<endl
#define debug2(x,y) cerr<< #x <<": "<< x <<", "<< #y <<": "<< y <<endl

using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> pii;
typedef vector<int> vi;
typedef vector<vector<int> > vvi;
typedef vector<ll> vll;
typedef vector<vector<ll> > vvll;
template<class T> using vv=vector<vector< T > >;

struct UF {
  vector<int> par; // parent
  vector<int> sizes;
  vi odd_edge;
  deque<bool> include_odd;
  UF(int n) : par(n), sizes(n, 1), odd_edge(n, -1), include_odd(n, false) {
    for (int i = 0; i < n; ++i) {
      par[i] = i;
    }
  }
  int root(int x) {
    if (x == par[x]) {
      return x;
    }
    return par[x] = root(par[x]);
  }
  void unite(int x, int y) {
    x = root(x);
    y = root(y);
    if (x == y) {
      return;
    }
    if (sizes[x] < sizes[y]) {
      swap(x, y);
    }
    par[y] = x;
    sizes[x] += sizes[y];
    sizes[y] = 0;
  }
  void unite_even(int x, int y) {
    x = root(x);
    y = root(y);
    if (x == y) {
      return;
    }
    if (odd_edge[x] == y && odd_edge[y] == x) {
      odd_edge[x] = -1;
      odd_edge[y] = -1;
      unite(x, y);
      x = root(x);
      include_odd[x] = true;
      return;
    }
    if (include_odd[x] || include_odd[y]) {
      if (include_odd[x]) {
        swap(x, y);
      }
      if (odd_edge[x] != -1) {
        int v = odd_edge[x];
        odd_edge[v] = -1;
        odd_edge[x] = -1;
        unite(x, y);
        unite(x, v);
        x = root(x);
        include_odd[x] = true;
        return;
      }
      unite(x, y);
      return;
    }
    if (odd_edge[x] != -1) {
      swap(x, y);
    }
    if (odd_edge[y] != -1) {
      int u = odd_edge[y];
      if (odd_edge[x] != -1) {
        int v = odd_edge[x];
        odd_edge[u] = -1;
        odd_edge[v] = -1;
        unite(u, v);
        odd_edge[x] = -1;
        odd_edge[y] = -1;
        unite(x, y);
        u = root(u);
        y = root(y);
        odd_edge[u] = y;
        odd_edge[y] = u;
        return;
      }
      odd_edge[y] = -1;
      unite(x, y);
      x = root(x);
      odd_edge[u] = x;
      odd_edge[x] = u;
      return;
    }
    unite(x, y);
    return;
  }
  void unite_odd(int x, int y) {
    x = root(x);
    y = root(y);
    if (x == y) {
      if (odd_edge[x] != -1) {
        int v = odd_edge[x];
        odd_edge[v] = -1;
        odd_edge[x] = -1;
        unite(x, v);
        x = root(x);
      }
      include_odd[x] = true;
      return;
    }
    if (odd_edge[x] == y && odd_edge[y] == x) {
      return;
    }
    if (include_odd[x] || include_odd[y]) {
      int u = -1;
      if (odd_edge[x] != -1) {
        u = x;
      } else if (odd_edge[y] != -1) {
        u = y;
      }
      if (u != -1) {
        int v = odd_edge[u];
        odd_edge[u] = -1;
        odd_edge[v] = -1;
        unite(u, v);
      }
      unite(x, y);
      x = root(x);
      include_odd[x] = true;
      return;
    }
    int u, v;
    u = v = -1;
    if (odd_edge[x] != -1) {
      swap(x, y);
    }
    if (odd_edge[y] != -1) {
      u = odd_edge[y];
      odd_edge[u] = -1;
      odd_edge[y] = -1;
      if (odd_edge[x] != -1) {
        v = odd_edge[x];
        odd_edge[v] = -1;
        odd_edge[x] = -1;
        unite(y, v);
        unite(x, u);
        u = root(u);
        v = root(v);
        odd_edge[u] = v;
        odd_edge[v] = u;
        return;
      }
      unite(x, u);
      x = root(x);
      odd_edge[x] = y;
      odd_edge[y] = x;
      return;
    }
    odd_edge[x] = y;
    odd_edge[y] = x;
    return;
  }
  void unite_main(int x, int y, int z) {
    if ((z&1) == 0) {
      unite_even(x, y);
    }
    else {
      unite_odd(x, y);
    }
    return;
  }
  bool same(int x, int y) {
    return root(x) == root(y);
  }
  int size(int x) {
    return sizes[root(x)];
  }
};

int main() {
  int n, q;
  scanf("%d %d", &n, &q);
  UF uf(n);
  int w, x, y, z;
  rep (query, q) {
    scanf("%d %d %d %d", &w, &x, &y, &z);
    x -= 1;
    y -= 1;
    if (w == 2) {
      if (uf.same(x, y)) {
        printf("YES\n");
      } else {
        printf("NO\n");
      }
      continue;
    } else {
      uf.unite_main(x, y, z);
    }
  }

  return 0;
}

Submission Info

Judge Result