結果

問題 No.2290 UnUnion Find
ユーザー prism17
提出日時 2023-07-19 15:47:23
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 154 ms / 2,000 ms
コード長 1,802 bytes
コンパイル時間 2,862 ms
コンパイル使用メモリ 176,728 KB
実行使用メモリ 14,976 KB
最終ジャッジ日時 2024-09-19 14:38:59
合計ジャッジ時間 12,943 ms
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

diff #

																		// Problem: No.2290 UnUnion Find No.2290 联合查找
// Contest: yukicoder
// URL: https://yukicoder.me/problems/no/2290
// Memory Limit: 512 MB
// Time Limit: 2000 ms

#include <bits/stdc++.h>

#define fastio ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
#define dbg(x) cout << #x << " = " << (x) << "\n";
#define popcount(x) __builtin_popcountll((x))
#define all(v) (v).begin(), (v).end()
#define pb emplace_back
#define x first
#define y second

using namespace std;
typedef long long ll;
typedef pair<ll, ll> pll;

const int inf = 0x3f3f3f3f;
const int mod = 1e9 + 7;

const int N = 2e5 + 7;
set<int> seq;

class dsu {
 private:
  vector<int> fa, sz, dis;

 public:
  void build(int N) {
    fa.resize(N + 1);
    sz.resize(N + 1);
    dis.resize(N + 1);
    for (int i = 0; i <= N; ++i) {
      fa[i] = i;
      sz[i] = 1;
    }
  }
  int find(int x) {
    if (fa[x] == x) return x;
    int root = find(fa[x]);
    dis[x] += dis[fa[x]];
    return fa[x] = root;
  }
  void join(int x, int y) {
    x = find(x), y = find(y);
    if (x != y) {
      fa[x] = y;
      sz[y] += sz[x];
      seq.erase(x);
    }
  }
  int size(int i) { return sz[find(i)]; }
  bool same(int x, int y) { return find(x) == find(y); }
};

void solve() {
  int n, q;
  cin >> n >> q;
  dsu d;
  d.build(n);
  for (int i = 1; i <= n; i++) seq.insert(i);
  for (int i = 1; i <= q; i++) {
    int op;
    cin >> op;
    if (op == 1) {
      int x, y;
      cin >> x >> y;
      d.join(x, y);
    } else {
      int x;
      cin >> x;
      int cur = -1;
      for (auto &p : seq) {
        if (!d.same(x, p)) {
          cur = p;
          break;
        }
      }
      cout << cur << "\n";
    }
  }
}

int main() {
  fastio;
  int t = 1;
  // cin >> t;
  while (t--) {
    solve();
  }
  return 0;
}
0