結果

問題 No.1054 Union add query
コンテスト
ユーザー i_am_nicolen_22
提出日時 2020-05-16 01:19:49
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 714 ms / 2,000 ms
コード長 2,966 bytes
コンパイル時間 1,862 ms
コンパイル使用メモリ 173,716 KB
実行使用メモリ 40,596 KB
最終ジャッジ日時 2024-09-19 16:49:01
合計ジャッジ時間 7,280 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
 
#define REP(i, s) for (int i = 0; i < s; ++i)
#define ALL(v) (v).begin(), (v).end()
#define COUT(x) cout << #x << " = " << (x) << " (L" << __LINE__ << ")" << endl
#define EACH(i, s) for (__typeof__((s).begin()) i = (s).begin(); i != (s).end(); ++i)
#define DEBUG
#define int long long
#define INF 1e18
template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; }
template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; }
template<class T1, class T2> ostream& operator << (ostream &s, pair<T1,T2> P)
{ return s << '<' << P.first << ", " << P.second << '>'; }
template<class T> ostream& operator << (ostream &s, vector<T> P)
{ for (int i = 0; i < P.size(); ++i) { if (i > 0) { s << " "; } s << P[i]; } return s; }
template<class T> ostream& operator << (ostream &s, vector<vector<T> > P)
{ for (int i = 0; i < P.size(); ++i) { s << endl << P[i]; } return s << endl; }
template<class T> ostream& operator << (ostream &s, set<T> P)
{ EACH(it, P) { s << "<" << *it << "> "; } return s << endl; }
template<class T1, class T2> ostream& operator << (ostream &s, map<T1,T2> P)
{ EACH(it, P) { s << "<" << it->first << "->" << it->second << "> "; } return s << endl; }
template<class T>void show(vector<T>v){for (int i = 0; i < v.size(); i++){cerr<<v[i]<<" ";}cerr<<"\n";}
typedef long long ll;

struct UnionFind
{
    vector<int> par,val;
    vector<vector<int>> child;

    UnionFind(int n) : par(n, -1) ,val(n) , child(n) {
        REP(i,n) {
            child[i].push_back(i);
        }
    }
    void init(int n) { par.assign(n, -1); }

    int root(int x)
    {
        if (par[x] < 0)
            return x;
        return par[x] = root(par[x]);
    }

    bool issame(int x, int y)
    {
        return root(x) == root(y);
    }
    bool merge(int x, int y)
    {
        x = root(x);
        y = root(y);
        if (x == y)
            return false;
        if (par[x] > par[y])
            swap(x, y);
        par[x] += par[y];
        par[y] = x; //大きさが正になるよ
        int sub = val[y] - val[x];
        val[y] = 0;
        for(auto & u : child[y]){
            child[x].push_back(u);
            val[u] += sub;
        }
        //child[y].clear();
        return true;
    }
    void add( int a, int b ){
        val[root(a)] += b;
    }
    ll get (ll a){
        if(par[a] < 0)
            return val[a];
        return val[a] + val[root(a)];
    }
    int size(int x)
    {
        return -par[root(x)];
    }
};

signed main()
{
    int n, q;
    cin >> n >> q;
    UnionFind uni(n);
    REP(i,q){
        int t, a, b;
        cin >> t >> a >> b;
        if(t==1){
            a--, b--;
            uni.merge(a, b);
        }
        else if(t==2){
            a--;
            uni.add(a, b);
        }
        else{
            a--;
            int x = uni.get(a);
            cout << x << endl;
        }
    }
    return 0;
}
0