結果

問題 No.1451 集団登校
ユーザー 👑 Nachia
提出日時 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--;
      |              ~~~~~^~~~~~~~~~~~~~

ソースコード

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