結果

問題 No.1054 Union add query
ユーザー beetbeet
提出日時 2020-05-15 21:32:37
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 217 ms / 2,000 ms
コード長 1,629 bytes
コンパイル時間 2,475 ms
コンパイル使用メモリ 204,996 KB
実行使用メモリ 11,008 KB
最終ジャッジ日時 2024-09-19 08:50:04
合計ジャッジ時間 6,356 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 148 ms
5,376 KB
testcase_04 AC 217 ms
10,880 KB
testcase_05 AC 135 ms
5,376 KB
testcase_06 AC 145 ms
6,272 KB
testcase_07 AC 124 ms
6,400 KB
testcase_08 AC 140 ms
6,400 KB
testcase_09 AC 181 ms
10,880 KB
testcase_10 AC 104 ms
11,008 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
template<typename T1,typename T2> inline void chmin(T1 &a,T2 b){if(a>b) a=b;}
template<typename T1,typename T2> inline void chmax(T1 &a,T2 b){if(a<b) a=b;}
using Int = long long;
const char newl = '\n';


template <typename T>
struct WeightedUnionFind{
  vector<int> rs,ps;
  vector<T> ws;

  WeightedUnionFind(){}
  WeightedUnionFind(int n):
    rs(n,1),ps(n),ws(n,T(0)){iota(ps.begin(),ps.end(),0);}

  int find(int x){
    if(x==ps[x]) return x;
    int t=find(ps[x]);
    ws[x]+=ws[ps[x]];
    return ps[x]=t;
  }

  T weight(int x){
    find(x);
    return ws[x];
  }

  bool same(int x,int y){
    return find(x)==find(y);
  }

  void unite(int x,int y,T w){
    w+=weight(x);
    w-=weight(y);
    x=find(x);y=find(y);
    if(x==y) return;
    if(rs[x]<rs[y]) swap(x,y),w=-w;
    rs[x]+=rs[y];
    ps[y]=x;
    ws[y]=w;
  }

  T diff(int x,int y){
    return weight(y)-weight(x);
  }
};

//INSERT ABOVE HERE
signed main(){
  cin.tie(0);
  ios::sync_with_stdio(0);

  int n,q;
  cin>>n>>q;
  WeightedUnionFind<int> uf(n);

  vector<int> vs(n,0);
  auto add=[&](int a,int b){
    vs[uf.find(a)]+=b;
  };
  auto query=[&](int a){
    int r=uf.find(a);
    return uf.diff(a,r)+vs[r];
  };
  auto unite=[&](int a,int b){
    a=uf.find(a);
    b=uf.find(b);
    if(a==b) return;
    int x=query(a);
    int y=query(b);
    uf.unite(a,b,x-y);
  };

  for(int i=0;i<q;i++){
    int t,a,b;
    cin>>t>>a>>b;
    if(t==1){
      a--;b--;
      unite(a,b);
    }
    if(t==2){
      a--;
      add(a,b);
    }
    if(t==3){
      a--;
      cout<<query(a)<<newl;
    }
  }
  return 0;
}
0