結果

問題 No.1054 Union add query
ユーザー parukiparuki
提出日時 2021-07-05 12:32:38
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 343 ms / 2,000 ms
コード長 2,388 bytes
コンパイル時間 2,280 ms
コンパイル使用メモリ 212,284 KB
実行使用メモリ 42,296 KB
最終ジャッジ日時 2023-09-14 04:09:40
合計ジャッジ時間 6,103 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,380 KB
testcase_01 AC 1 ms
4,376 KB
testcase_02 AC 1 ms
4,380 KB
testcase_03 AC 180 ms
11,780 KB
testcase_04 AC 343 ms
42,108 KB
testcase_05 AC 155 ms
7,652 KB
testcase_06 AC 162 ms
18,492 KB
testcase_07 AC 151 ms
18,544 KB
testcase_08 AC 159 ms
19,124 KB
testcase_09 AC 222 ms
42,296 KB
testcase_10 AC 145 ms
42,172 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define _USE_MATH_DEFINES
#include "bits/stdc++.h"
using namespace std;
#define FOR(i,j,k) for(int (i)=(j);(i)<(int)(k);++(i))
#define rep(i,j) FOR(i,0,j)
#define each(x,y) for(auto &(x):(y))
#define mp make_pair
#define MT make_tuple
#define all(x) (x).begin(),(x).end()
#define debug(x) cout<<#x<<": "<<(x)<<endl
#define smax(x,y) (x)=max((x),(y))
#define smin(x,y) (x)=min((x),(y))
#define MEM(x,y) memset((x),(y),sizeof (x))
#define sz(x) (int)(x).size()
#define RT return
using ll = long long;
using pii = pair<int, int>;
using vi = vector<int>;
using vll = vector<ll>;

class UnionFindBySize {
    int cnt;
    vector<int> par, size;
public:
    UnionFindBySize() :cnt(0) {}
    UnionFindBySize(int _n) :cnt(_n), par(_n), size(_n, 1) {
        for (int i = 0; i < _n; ++i) par[i] = i;
    }
    int find(int k) {
        return (k == par[k]) ? k : (par[k] = find(par[k]));
    }
    int operator[](int k) {
        return find(k);
    }
    int getSize(int k) {
        return size[find(k)];
    }
    void unite(int x, int y) {
        x = find(x); y = find(y);
        if (x == y) return;
        --cnt;
        if (size[x] < size[y]) {
            par[x] = y;
            size[y] += size[x];
        } else {
            par[y] = x;
            size[x] += size[y];
        }
    }
    int count() {
        return cnt;
    }
};

void solve() {
    int N, Q;
    scanf("%d %d", &N, &Q);
    UnionFindBySize UF(N);
    vector<vi> V(N);
    rep(i, N) {
        V[i].push_back(i);
    }
    vll X(N), Y(N); // A[i] = X[i]+Y[UF[i]]
    rep(ITER, Q) {
        int T, A, B;
        scanf("%d %d %d", &T, &A, &B);
        --A;
        if (T == 1) {
            --B;
            if (UF[A] == UF[B]) {
                continue;
            }
            if (UF.getSize(A) < UF.getSize(B)) {
                swap(A, B);
            }
            int u = UF[A], v = UF[B];
            UF.unite(u, v);
            each(a, V[v]) {
                X[a] += Y[v] - Y[u];
            }
            ll y = Y[u];
            V[u].insert(V[u].end(), all(V[v]));
            V[v].clear();
            V[v].shrink_to_fit();
        } else if (T == 2) {
            Y[UF[A]] += B;
        } else {
            printf("%lld\n", X[A] + Y[UF[A]]);
        }
    }
}

int main() {
    // ios::sync_with_stdio(false);
    // cin.tie(0);
    // cout << fixed << setprecision(15);
    solve();
    return 0;
}
0