結果

問題 No.1451 集団登校
ユーザー 👑 NachiaNachia
提出日時 2021-03-31 15:16:10
言語 C++17
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 34 ms / 2,000 ms
コード長 1,203 bytes
コンパイル時間 2,128 ms
コンパイル使用メモリ 206,164 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2023-08-21 12:19:20
合計ジャッジ時間 4,245 ms
ジャッジサーバーID
(参考情報)
judge15 / judge13
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,380 KB
testcase_01 AC 2 ms
4,380 KB
testcase_02 AC 1 ms
4,376 KB
testcase_03 AC 1 ms
4,376 KB
testcase_04 AC 2 ms
4,380 KB
testcase_05 AC 1 ms
4,376 KB
testcase_06 AC 2 ms
4,380 KB
testcase_07 AC 1 ms
4,376 KB
testcase_08 AC 22 ms
4,640 KB
testcase_09 AC 1 ms
4,376 KB
testcase_10 AC 16 ms
5,212 KB
testcase_11 AC 11 ms
5,376 KB
testcase_12 AC 16 ms
4,840 KB
testcase_13 AC 26 ms
5,364 KB
testcase_14 AC 33 ms
4,972 KB
testcase_15 AC 16 ms
4,452 KB
testcase_16 AC 23 ms
4,980 KB
testcase_17 AC 18 ms
4,380 KB
testcase_18 AC 10 ms
4,380 KB
testcase_19 AC 8 ms
4,424 KB
testcase_20 AC 34 ms
4,972 KB
testcase_21 AC 12 ms
4,376 KB
testcase_22 AC 8 ms
4,376 KB
testcase_23 AC 4 ms
4,376 KB
testcase_24 AC 15 ms
5,196 KB
testcase_25 AC 16 ms
4,380 KB
testcase_26 AC 23 ms
4,980 KB
testcase_27 AC 24 ms
4,376 KB
testcase_28 AC 9 ms
4,376 KB
testcase_29 AC 16 ms
4,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
using ll=long long;
using ull=unsigned long long;
#define rep(i,n) for(int i=0; i<(n); i++)

const ull M = 1000000007;
const ull inv2 = M/2+1;

struct DSU{
  vector<int> V;
  vector<int> Z;
  vector<ull> P;
  vector<ull> iP;
  DSU(int n=0){
    V.assign(n,-1);
    Z.assign(n,1);
    P.assign(n,1);
    iP.assign(n,1);
  }
  pair<int,ull> leader(int a){
    if(V[a]>=0){
      auto r=leader(V[a]);
      V[a]=r.first;
      P[a]=P[a]*r.second%M;
      return {V[a],P[a]};
    }
    return {a,1};
  }
  ull value(int a){
    auto ap = leader(a);
    return ap.second * P[ap.first] % M;
  }
  void merge(int u,int v){
    auto up = leader(u); u=up.first;
    auto vp = leader(v); v=vp.first;
    if(u==v) return;
    if(Z[u]<Z[v]) swap(u,v);
    if(Z[u]>Z[v]){
      P[v]=0; iP[v]=0;
      V[v]=u; Z[u]+=Z[v];
    }
    else{
      P[v]=P[v]*iP[u]%M; iP[v]=iP[v]*P[u]%M;
      P[u]=P[u]*inv2%M; iP[u]=iP[u]*2%M;
      V[v]=u; Z[u]+=Z[v];
    }
  }
};

int N,Q;
DSU dsu;

int main(){
  scanf("%d%d",&N,&Q);
  dsu = DSU(N);
  rep(i,Q){
    int u,v; scanf("%d%d",&u,&v); u--; v--;
    dsu.merge(u,v);
  }
  rep(i,N) printf("%llu\n",dsu.value(i));
  return 0;
}
0