結果

問題 No.2290 UnUnion Find
コンテスト
ユーザー ting shuo
提出日時 2026-08-26 01:15:34
言語 C++17
(gcc 15.2.0 + boost 1.90.0)
コンパイル:
g++-15 -O2 -lm -std=c++17 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 136 ms / 2,000 ms
+ 181µs
コード長 1,631 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,302 ms
コンパイル使用メモリ 218,560 KB
実行使用メモリ 13,568 KB
最終ジャッジ日時 2026-08-26 01:15:50
合計ジャッジ時間 9,650 ms
ジャッジサーバーID
(参考情報)
judge2_0 / judge1_0
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

#include <bits/stdc++.h>

using i64 = long long;
using u64 = unsigned long long;
using u32 = unsigned;

using u128 = unsigned __int128;
using i128 = __int128;

std::set<int> roots;

struct DSU {
    std::vector<int> f;
    DSU() {}
    DSU(int n) {
        init(n);
    }
    void init(int n) {
        f.resize(n);
        std::iota(f.begin(), f.end(), 0);
    }
    int find(int x) {
        while (x != f[x]) {
            x = f[x] = f[f[x]];
        }
        return x;
    }
    bool merge(int x, int y) {
        x = find(x);
        y = find(y);
        if (x == y) {
            return false;
        }
        f[y] = x;
        roots.erase(y);
        return true;
    }
    int get_root(int x) {
        return find(x);
    }
};

int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    
    int N, Q;
    std::cin >> N >> Q;
    
    for (int i = 0; i < N; i++) {
        roots.insert(i);
    }
    DSU dsu(N);

    while (Q--) {
        int op;
        std::cin >> op;
        if (op == 1) {
            int u, v;
            std::cin >> u >> v;
            u--;
            v--;
            dsu.merge(u, v);
        } else {
            int u;
            std::cin >> u;
            u--;
            int fa = dsu.get_root(u);
            if (roots.size() == 1) {
                std::cout << -1 << '\n';
            } else {
                auto it = roots.begin();
                if (*it == fa) {
                    std::cout << *(++it) + 1 << '\n';
                } else {
                    std::cout << *it + 1 << '\n';
                }
            }
        }
    }

    return 0;
}
0