結果

問題 No.1054 Union add query
ユーザー koyoprokoyopro
提出日時 2020-05-15 23:09:52
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 311 ms / 2,000 ms
コード長 2,434 bytes
コンパイル時間 1,507 ms
コンパイル使用メモリ 168,520 KB
実行使用メモリ 44,632 KB
最終ジャッジ日時 2024-09-19 13:11:51
合計ジャッジ時間 4,439 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,812 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 1 ms
6,944 KB
testcase_03 AC 169 ms
13,456 KB
testcase_04 AC 311 ms
44,632 KB
testcase_05 AC 143 ms
8,320 KB
testcase_06 AC 147 ms
20,852 KB
testcase_07 AC 129 ms
20,980 KB
testcase_08 AC 147 ms
20,980 KB
testcase_09 AC 221 ms
42,400 KB
testcase_10 AC 131 ms
42,256 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
using namespace std;
#define int long long
#define FOR(i, a, b) for(int i=(a);i<(b);i++)
#define RFOR(i, a, b) for(int i=(b-1);i>=(a);i--)
#define REP(i, n) for(int i=0; i<(n); i++)
#define RREP(i, n) for(int i=(n-1); i>=0; i--)
#define REP1(i, n) for(int i=1; i<=(n); i++)
#define RREP1(i, n) for(int i=(n); i>=1; i--)
#define ALL(a) (a).begin(),(a).end()
#define UNIQUE_SORT(l) sort(ALL(l)); l.erase(unique(ALL(l)), l.end());
#define CONTAIN(a, b) find(ALL(a), (b)) != (a).end()
#define out(...) printf(__VA_ARGS__)

int dxy[] = {0, 1, 0, -1, 0};

void solve();
signed main()
{
#if DEBUG
    std::ifstream in("input.txt");
    std::cin.rdbuf(in.rdbuf());
#endif
    cin.tie(0);
    ios::sync_with_stdio(false);
    solve();
    return 0;
}

/*================================*/
#if DEBUG
#define SIZE 100
#else
#define SIZE 123450
#endif

int N,M,Q;

struct UnionFind {
    vector<int> parent;
    vector<int> A; // 個別
    vector<int> B; // 全体
    vector<vector<int>> vec;
    int size;
    UnionFind(int size):size(size) {
        reset(size);
    }
    void reset(int size) {
        this->size = size;
        parent.resize(size);
        A.resize(size, 0);
        B.resize(size, 0);
        vec.resize(size);
        REP(i,size) parent[i]=i;
        REP(i,size) vec[i].push_back(i);
    }
    int root(int i) {
        if (parent[i] == i) return i;
        return parent[i] = root(parent[i]);
    }
    void merge(int i, int j) {
        int ir = root(i);
        int jr = root(j);
        if (ir == jr) return;
        int next = vec[ir].size() >= vec[jr].size() ? ir : jr;
        int hide = (ir == next) ? jr : ir;
        // hideの中身をnextにマージ
        for (int i:vec[hide]) {
            A[i] += B[hide] - B[next]; // 差分更新
            vec[next].push_back(i);
        }
        vec[hide].clear();
        parent[ir] = parent[jr] = next;
    }
    bool same(int i, int j) {
        return root(i) == root(j);
    }
    void add(int i, int a) {
        B[root(i)] += a;
    }
    int val(int i) {
        return A[i] + B[root(i)];
    }
};

void solve() {
    cin>>N>>Q;
    int T,A,B;
    auto uf = UnionFind(N);
    REP(q,Q) {
        cin>>T>>A>>B;
        A--;
        if (T==1) {
            B--;
            uf.merge(A, B);
        }
        if (T==2) {
            uf.add(A, B);
        }
        if (T==3) {
            out("%lld\n", uf.val(A));
        }
    }
}
0