結果
問題 | No.1451 集団登校 |
ユーザー |
👑 ![]() |
提出日時 | 2021-03-31 15:16:10 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 34 ms / 2,000 ms |
コード長 | 1,203 bytes |
コンパイル時間 | 2,031 ms |
コンパイル使用メモリ | 201,704 KB |
最終ジャッジ日時 | 2025-01-20 01:49:45 |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 27 |
コンパイルメッセージ
main.cpp: In function ‘int main()’: main.cpp:56:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result] 56 | scanf("%d%d",&N,&Q); | ~~~~~^~~~~~~~~~~~~~ main.cpp:59:19: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result] 59 | int u,v; scanf("%d%d",&u,&v); u--; v--; | ~~~~~^~~~~~~~~~~~~~
ソースコード
#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; }