Problem p02540 (C++) — buggy submission, verdict: Time Limit Exceeded

/*
このコード、と～おれ!
Be accepted!
∧＿∧　
（｡･ω･｡)つ━☆・*。
⊂　　 ノ 　　　・゜+.
　しーＪ　　　°。+ *´¨)
 　　　　　　　　　.· ´¸.·*´¨) ¸.·*¨)
                  　　　　　　　　　　(¸.·´ (¸.·'* ☆
*/

#include <algorithm>
#include <bitset>
#include <cassert>
#include <cmath>
#include <complex>
#include <cstdio>
#include <cstring>
#include <functional>
#include <iomanip>
#include <iostream>
#include <list>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <regex>
#include <set>
#include <stack>
#include <string>
#include <unordered_map>
#include <vector>

////多倍長整数, cpp_intで宣言
// #include <boost/multiprecision/cpp_int.hpp>
// using namespace boost::multiprecision;
//
// #pragma GCC target ("avx2")
// #pragma GCC optimization ("Ofast")
// #pragma GCC optimization ("unroll-loops")
// #pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
#define repeat(i, n, m) for (int i = n; i < (m); ++i)
#define rep(i, n) for (int i = 0; i < (n); ++i)
#define printynl(a) printf(a ? "yes\\n" : "no\\n")
#define printyn(a) printf(a ? "Yes\\n" : "No\\n")
#define printYN(a) printf(a ? "YES\\n" : "NO\\n")
#define printim(a) printf(a ? "possible\\n" : "imposible\\n")
#define printdb(a) printf("%.50lf\\n", a)    // 少数出力
#define printLdb(a) printf("%.50Lf\\n", a)   // 少数出力
#define printdbd(a) printf("%.16lf\\n", a)   // 少数出力(桁少なめ)
#define prints(s) printf("%s\\n", s.c_str()) // string出力
#define all(x) (x).begin(), (x).end()
#define deg_to_rad(deg) (((deg) / 360.0L) * 2.0L * PI)
#define rad_to_deg(rad) (((rad) / 2.0L / PI) * 360.0L)
#define Please return
#define AC 0
#define manhattan_dist(a, b, c, d)                                             \\
  (abs(a - c) + abs(b - d)) /*(a, b) から (c, d) のマンハッタン距離 */
#define inf numeric_limits<double>::infinity();
#define linf numeric_limits<long double>::infinity()

using ll = long long;
using ull = unsigned long long;

constexpr int INF = 1073741823;
constexpr int MINF = -1073741823;
constexpr ll LINF = ll(4661686018427387903);
constexpr ll MOD = 1e9 + 7;
constexpr ll mod = 998244353;
constexpr long double eps = 1e-6;
const long double PI = acosl(-1.0L);

using namespace std;

void scans(string &str) {
  char c;
  str = "";
  scanf("%c", &c);
  if (c == '\\n')
    scanf("%c", &c);
  while (c != '\\n' && c != -1 && c != ' ') {
    str += c;
    scanf("%c", &c);
  }
}

void scanc(char &str) {
  char c;
  scanf("%c", &c);
  if (c == -1)
    return;
  while (c == '\\n') {
    scanf("%c", &c);
  }
  str = c;
}

double acot(double x) { return PI / 2 - atan(x); }

ll LSB(ll n) { return (n & (-n)); }

template <typename T> inline T chmin(T &a, const T &b) {
  if (a > b)
    a = b;
  return a;
}

template <typename T> inline T chmax(T &a, const T &b) {
  if (a < b)
    a = b;
  return a;
}

////atcoder library
// #include <atcoder/all>
// using namespace atcoder;

/*-----------------------------------------ここからコード-----------------------------------------*/

/*
 * @title unionfind
 * @docs kyopro/docs/unionfind.md
 */

// 0-indexed
struct unionfind {
  vector<int> par, siz;

  unionfind(int n) : par(n), siz(n) {
    for (int i = 0; i < n; ++i) {
      // 全部根で初期化
      par[i] = i;
      // サイズは1
      siz[i] = 1;
    }
  }

  void init(int n) {
    par.resize(n);
    siz.resize(n);
    for (int i = 0; i < n; ++i) {
      // 全部根で初期化
      par[i] = i;
      // サイズは1
      siz[i] = 1;
    }
  }

  // 根を返す
  int find(int a) { return par[a] == a ? a : par[a] = find(par[a]); }

  // くっつける。元から同じだったらfalseを返す
  bool unite(int a, int b) {
    int x = find(a), y = find(b);
    if (x == y)
      return false;
    else if (siz[x] < siz[y]) {
      par[x] = y;
      siz[y] += siz[x];
    } else if (siz[x] > siz[y]) {
      par[y] = x;
      siz[x] += siz[y];
    } else {
      par[x] = y;
      siz[y] += siz[x];
    }
    return true;
  }

  // 同じ集合か判定する
  bool same(int a, int b) { return find(a) == find(b); }

  // サイズを返す
  int size(int a) { return siz[find(a)]; }

  // 同じ集合に属す葉を纏めて返す
  vector<int> leaf(int a) {
    vector<int> x;
    int n = par.size();
    for (int i = 0; i < n; ++i)
      if (same(a, i))
        x.push_back(i);
    return x;
  }
};

int main() {

  int n;
  scanf("%d", &n);
  int cnt = 0;
  unionfind uni(n);
  set<pair<int, int>> st;
  vector<tuple<int, int, int>> xy(n);
  for (auto &[a, b, qu] : xy) {
    scanf("%d%d", &a, &b);
    --a, --b;
    qu = cnt++;
  }
  sort(all(xy));
  rep(i, n) {
    int nm = get<1>(xy[i]);
    for (auto ite = st.begin();
         ite != st.end() and ite->first < get<1>(xy[i]);) {
      uni.unite(ite->second, get<2>(xy[i]));
      chmin(nm, ite->first);
      ite = st.erase(ite);
    }
    st.insert({nm, get<2>(xy[i])});
  }
  rep(i, n) printf("%d\\n", uni.size(i));

  Please AC;
}

Error: TLE

Fixed submission:

/*
このコード、と～おれ!
Be accepted!
∧＿∧　
（｡･ω･｡)つ━☆・*。
⊂　　 ノ 　　　・゜+.
　しーＪ　　　°。+ *´¨)
 　　　　　　　　　.· ´¸.·*´¨) ¸.·*¨)
                  　　　　　　　　　　(¸.·´ (¸.·'* ☆
*/

#include <algorithm>
#include <bitset>
#include <cassert>
#include <cmath>
#include <complex>
#include <cstdio>
#include <cstring>
#include <functional>
#include <iomanip>
#include <iostream>
#include <list>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <regex>
#include <set>
#include <stack>
#include <string>
#include <unordered_map>
#include <vector>

////多倍長整数, cpp_intで宣言
// #include <boost/multiprecision/cpp_int.hpp>
// using namespace boost::multiprecision;
//
#pragma GCC target("avx2")
#pragma GCC optimization("Ofast")
#pragma GCC optimization("unroll-loops")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
#define repeat(i, n, m) for (int i = n; i < (m); ++i)
#define rep(i, n) for (int i = 0; i < (n); ++i)
#define printynl(a) printf(a ? "yes\\n" : "no\\n")
#define printyn(a) printf(a ? "Yes\\n" : "No\\n")
#define printYN(a) printf(a ? "YES\\n" : "NO\\n")
#define printim(a) printf(a ? "possible\\n" : "imposible\\n")
#define printdb(a) printf("%.50lf\\n", a)    // 少数出力
#define printLdb(a) printf("%.50Lf\\n", a)   // 少数出力
#define printdbd(a) printf("%.16lf\\n", a)   // 少数出力(桁少なめ)
#define prints(s) printf("%s\\n", s.c_str()) // string出力
#define all(x) (x).begin(), (x).end()
#define deg_to_rad(deg) (((deg) / 360.0L) * 2.0L * PI)
#define rad_to_deg(rad) (((rad) / 2.0L / PI) * 360.0L)
#define Please return
#define AC 0
#define manhattan_dist(a, b, c, d)                                             \\
  (abs(a - c) + abs(b - d)) /*(a, b) から (c, d) のマンハッタン距離 */
#define inf numeric_limits<double>::infinity();
#define linf numeric_limits<long double>::infinity()

using ll = long long;
using ull = unsigned long long;

constexpr int INF = 1073741823;
constexpr int MINF = -1073741823;
constexpr ll LINF = ll(4661686018427387903);
constexpr ll MOD = 1e9 + 7;
constexpr ll mod = 998244353;
constexpr long double eps = 1e-6;
const long double PI = acosl(-1.0L);

using namespace std;

void scans(string &str) {
  char c;
  str = "";
  scanf("%c", &c);
  if (c == '\\n')
    scanf("%c", &c);
  while (c != '\\n' && c != -1 && c != ' ') {
    str += c;
    scanf("%c", &c);
  }
}

void scanc(char &str) {
  char c;
  scanf("%c", &c);
  if (c == -1)
    return;
  while (c == '\\n') {
    scanf("%c", &c);
  }
  str = c;
}

double acot(double x) { return PI / 2 - atan(x); }

ll LSB(ll n) { return (n & (-n)); }

template <typename T> inline T chmin(T &a, const T &b) {
  if (a > b)
    a = b;
  return a;
}

template <typename T> inline T chmax(T &a, const T &b) {
  if (a < b)
    a = b;
  return a;
}

////atcoder library
// #include <atcoder/all>
// using namespace atcoder;

/*-----------------------------------------ここからコード-----------------------------------------*/

/*
 * @title unionfind
 * @docs kyopro/docs/unionfind.md
 */

// 0-indexed
struct unionfind {
  vector<int> par, siz;

  unionfind(int n) : par(n), siz(n) {
    for (int i = 0; i < n; ++i) {
      // 全部根で初期化
      par[i] = i;
      // サイズは1
      siz[i] = 1;
    }
  }

  void init(int n) {
    par.resize(n);
    siz.resize(n);
    for (int i = 0; i < n; ++i) {
      // 全部根で初期化
      par[i] = i;
      // サイズは1
      siz[i] = 1;
    }
  }

  // 根を返す
  int find(int a) { return par[a] == a ? a : par[a] = find(par[a]); }

  // くっつける。元から同じだったらfalseを返す
  bool unite(int a, int b) {
    int x = find(a), y = find(b);
    if (x == y)
      return false;
    else if (siz[x] < siz[y]) {
      par[x] = y;
      siz[y] += siz[x];
    } else if (siz[x] > siz[y]) {
      par[y] = x;
      siz[x] += siz[y];
    } else {
      par[x] = y;
      siz[y] += siz[x];
    }
    return true;
  }

  // 同じ集合か判定する
  bool same(int a, int b) { return find(a) == find(b); }

  // サイズを返す
  int size(int a) { return siz[find(a)]; }

  // 同じ集合に属す葉を纏めて返す
  vector<int> leaf(int a) {
    vector<int> x;
    int n = par.size();
    for (int i = 0; i < n; ++i)
      if (same(a, i))
        x.push_back(i);
    return x;
  }
};

int main() {

  int n;
  scanf("%d", &n);
  int cnt = 0, flag = 0;
  queue<tuple<int, int, int>> que;
  vector<bool> check(n);
  unionfind uni(n);
  vector<tuple<int, int, int>> xy(n);
  for (auto &[a, b, qu] : xy) {
    scanf("%d%d", &a, &b);
    --a, --b;
    qu = cnt++;
  }
  rep(i, n) {
    if (check[get<2>(xy[i])])
      continue;
    check[get<2>(xy[i])] = true;
    ++flag;
    que.push(xy[i]);
    while (not que.empty()) {
      auto [x, y, q] = que.front();
      que.pop();
      rep(j, n) {
        if (check[j])
          continue;
        else if ((x > get<0>(xy[j]) and y > get<1>(xy[j])) or
                 (x < get<0>(xy[j]) and y < get<1>(xy[j]))) {
          check[get<2>(xy[j])] = true;
          que.push(xy[j]);
          uni.unite(i, j);
          ++flag;
        }
      }
    }
    if (flag == n)
      break;
  }
  rep(i, n) printf("%d\\n", uni.size(i));

  Please AC;
}