結果

問題 No.1451 集団登校
コンテスト
ユーザー 👑 Nachia
提出日時 2021-03-31 15:12:01
言語 C++17(gcc12)
(gcc 12.4.0 + boost 1.89.0)
コンパイル:
g++-12 -O2 -lm -std=c++17 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
MLE  
実行時間 -
コード長 1,182 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 3,039 ms
コンパイル使用メモリ 208,632 KB
実行使用メモリ 1,307,008 KB
最終ジャッジ日時 2026-06-19 03:16:19
合計ジャッジ時間 4,948 ms
ジャッジサーバーID
（参考情報） 		judge1_0 / judge2_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 10 MLE * 2 -- * 15
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:55:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   55 |   scanf("%d%d",&N,&Q);
      |   ~~~~~^~~~~~~~~~~~~~
main.cpp:58:19: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   58 |     int u,v; scanf("%d%d",&u,&v); u--; v--;
      |              ~~~~~^~~~~~~~~~~~~~

ソースコード

diff #
raw source code 

#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(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