結果

問題 No.1054 Union add query
コンテスト
ユーザー 👑 tttmmmyyyy
提出日時 2026-10-31 08:19:29
言語 C++23
(gcc 15.3.0 + boost 1.92.0 + ACL)
コンパイル:
g++-15 -O2 -fplugin=/usr/lib/yukicoder/noavx512-gcc15.so -lm -std=c++23 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 229 ms / 2,000 ms
+ 546µs
コード長 1,216 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 450 ms
コンパイル使用メモリ 86,548 KB
実行使用メモリ 43,276 KB
最終ジャッジ日時 2026-10-31 08:19:39
合計ジャッジ時間 5,452 ms
ジャッジサーバーID
（参考情報） 		judge4_0 / judge3_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 8
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <cstdio>
#include <utility>
#include <vector>

using namespace std;
using ll = long long;

int main() {
    ll n, q;
    scanf("%lld %lld", &n, &q);
    vector<vector<ll>> gs(n);
    for (ll i = 0; i < n; i++) {
        gs[i] = {i};
    }
    vector<ll> gn(n, 0), vn(n, 0), v2g(n);
    for (ll i = 0; i < n; i++) {
        v2g[i] = i;
    }
    for (ll k = 0; k < q; k++) {
        ll t, a, b;
        scanf("%lld %lld %lld", &t, &a, &b);
        if (t == 1) {
            a--;
            b--;
            if (v2g[a] == v2g[b]) {
                continue;
            }
            if (!(gs[v2g[a]].size() < gs[v2g[b]].size())) {
                swap(a, b);
            }
            ll i = v2g[a], j = v2g[b];
            ll m_i = gn[i], m_j = gn[j];
            vector<ll> ga = move(gs[i]);
            gs[i] = {};
            for (ll v : ga) {
                v2g[v] = j;
            }
            for (ll v : ga) {
                vn[v] += m_i - m_j;
            }
            gs[j].insert(gs[j].end(), ga.begin(), ga.end());
        } else if (t == 2) {
            a--;
            gn[v2g[a]] += b;
        } else {
            a--;
            printf("%lld\n", vn[a] + gn[v2g[a]]);
        }
    }
}
0