Problem p03354 (C++)

Buggy submission (status: Runtime Error):
/*~=~=~=~=~=~=~=~=~=~=~=~=~=~=~=~=~=~=~=
*$* WRITER:kakitamasziru/OxOmisosiru *$*
~=~=~=~=~=~=~=~=~=~=~=~=~=~=~=~=~=~=~=*/
#ifdef LOCAL_JUDGE
#define _GLIBCXX_DEBUG
#endif
#include <algorithm> // min, max, swap, sort, reverse, lower_bound, upper_bound
#include <cstdint>   // int64_t, int*_t
#include <iomanip>
#include <iostream> // cout, endl, cin
#include <limits>   //setprecision
#include <string>   // string, to_string, stoi
#include <tuple>    // tuple, make_tuple
#include <utility>  // pair, make_pair
#include <vector>   // vector
// #include <cstdio> // printf
#include <map>   // map
#include <queue> // queue, priority_queue
#include <set>   // set
// #include <unordered_set> //unordered_set
#include <bitset> // bitset
#include <cmath>  //abs,,,
#include <deque>  // deque
#include <math.h> //pow,,,
#include <stack>  // stack
#define endl "\n";
// It is so troublesome that I include bits/stdc++.h !
using namespace std;

template <class T> inline bool chmin(T &a, T b) {
  if (a > b) {
    a = b;
    return true;
  }
  return false;
}
template <class T> inline bool chmax(T &a, T b) {
  if (a < b) {
    a = b;
    return true;
  }
  return false;
}
const long long INF = 100100100100100;
const long long MOD = 1000000007;
typedef pair<long long, int> P;

// Solve N^M. This, mod_pow use Iterative Square Method.
long long mod_pow(long long N, long long M) {
  if (M == 0)
    return 1;
  long long res = mod_pow((N * N) % MOD, M / 2);
  // 最下位ビット(*N)が1の時は単独でNをかける
  if (M & 1)
    res = (res * N) % MOD;

  return res %= MOD;
}

int gcd(long long a, long long b) {
  if (b == 0)
    return a;
  else
    return gcd(b, a % b);
}
long long lcm(long long a, long long b) { return a / gcd(a, b) * b; }

int parent[55];
int tall[55];

int searchRoot(int v) {
  if (parent[v] == v)
    return v;
  return parent[v] = searchRoot(parent[v]);
}

void unite(int x, int y) {
  int rootX = searchRoot(x);
  int rootY = searchRoot(y);
  if (rootX == rootY)
    return;
  if (tall[x] > tall[y]) {
    parent[rootY] = rootX;
  } else if (tall[x] < tall[y]) {
    parent[rootX] = rootY;
  } else {
    tall[y]++;
    parent[rootY] = rootX;
  }
}

bool same(int x, int y) {
  if (searchRoot(x) == searchRoot(y))
    return true;
  else
    return false;
}

int main() {
  ios::sync_with_stdio(false);
  std::cin.tie(nullptr);

  int N, M;
  cin >> N >> M;
  vector<int> retu(N);
  for (int i = 0; i < N; i++) {
    cin >> retu.at(i);
    retu.at(i)--;
  }
  for (int i = 0; i < N; i++) {
    parent[i] = i;
    tall[i] = 0;
  }
  // 連結しているグループ内に番号と一致する値があったら揃えられる
  for (int i = 0; i < M; i++) {
    int a, b;
    cin >> a >> b;
    a--;
    b--;
    unite(a, b);
  }
  // 連結しているグループごとに入れ替えられる値を保存する
  vector<set<int>> group(N);
  for (int i = 0; i < N; i++) {
    int root = searchRoot(i);
    group.at(root).insert(retu.at(i));
  }
  int ans = 0;
  // 属しているグループ内に同じ値があるか調べる
  for (int i = 0; i < N; i++) {
    int Root = searchRoot(i);
    if (group.at(Root).count(i))
      ans++;
  }

  cout << ans << endl;
}


Fixed submission:
/*~=~=~=~=~=~=~=~=~=~=~=~=~=~=~=~=~=~=~=
*$* WRITER:kakitamasziru/OxOmisosiru *$*
~=~=~=~=~=~=~=~=~=~=~=~=~=~=~=~=~=~=~=*/
#ifdef LOCAL_JUDGE
#define _GLIBCXX_DEBUG
#endif
#include <algorithm> // min, max, swap, sort, reverse, lower_bound, upper_bound
#include <cstdint>   // int64_t, int*_t
#include <iomanip>
#include <iostream> // cout, endl, cin
#include <limits>   //setprecision
#include <string>   // string, to_string, stoi
#include <tuple>    // tuple, make_tuple
#include <utility>  // pair, make_pair
#include <vector>   // vector
// #include <cstdio> // printf
#include <map>   // map
#include <queue> // queue, priority_queue
#include <set>   // set
// #include <unordered_set> //unordered_set
#include <bitset> // bitset
#include <cmath>  //abs,,,
#include <deque>  // deque
#include <math.h> //pow,,,
#include <stack>  // stack
#define endl "\n";
// It is so troublesome that I include bits/stdc++.h !
using namespace std;

template <class T> inline bool chmin(T &a, T b) {
  if (a > b) {
    a = b;
    return true;
  }
  return false;
}
template <class T> inline bool chmax(T &a, T b) {
  if (a < b) {
    a = b;
    return true;
  }
  return false;
}
const long long INF = 100100100100100;
const long long MOD = 1000000007;
typedef pair<long long, int> P;

// Solve N^M. This, mod_pow use Iterative Square Method.
long long mod_pow(long long N, long long M) {
  if (M == 0)
    return 1;
  long long res = mod_pow((N * N) % MOD, M / 2);
  // 最下位ビット(*N)が1の時は単独でNをかける
  if (M & 1)
    res = (res * N) % MOD;

  return res %= MOD;
}

int gcd(long long a, long long b) {
  if (b == 0)
    return a;
  else
    return gcd(b, a % b);
}
long long lcm(long long a, long long b) { return a / gcd(a, b) * b; }

int parent[100005];
int tall[100005];

int searchRoot(int v) {
  if (parent[v] == v)
    return v;
  return parent[v] = searchRoot(parent[v]);
}

void unite(int x, int y) {
  int rootX = searchRoot(x);
  int rootY = searchRoot(y);
  if (rootX == rootY)
    return;
  if (tall[x] > tall[y]) {
    parent[rootY] = rootX;
  } else if (tall[x] < tall[y]) {
    parent[rootX] = rootY;
  } else {
    tall[y]++;
    parent[rootY] = rootX;
  }
}

bool same(int x, int y) {
  if (searchRoot(x) == searchRoot(y))
    return true;
  else
    return false;
}

int main() {
  ios::sync_with_stdio(false);
  std::cin.tie(nullptr);

  int N, M;
  cin >> N >> M;
  vector<int> retu(N);
  for (int i = 0; i < N; i++) {
    cin >> retu.at(i);
    retu.at(i)--;
  }
  for (int i = 0; i < N; i++) {
    parent[i] = i;
    tall[i] = 0;
  }
  // 連結しているグループ内に番号と一致する値があったら揃えられる
  for (int i = 0; i < M; i++) {
    int a, b;
    cin >> a >> b;
    a--;
    b--;
    unite(a, b);
  }
  // 連結しているグループごとに入れ替えられる値を保存する
  vector<set<int>> group(N);
  for (int i = 0; i < N; i++) {
    int root = searchRoot(i);
    group.at(root).insert(retu.at(i));
  }
  int ans = 0;
  // 属しているグループ内に同じ値があるか調べる
  for (int i = 0; i < N; i++) {
    int Root = searchRoot(i);
    if (group.at(Root).count(i))
      ans++;
  }

  cout << ans << endl;
}