結果

問題 No.1451 集団登校
ユーザー shiomusubi496shiomusubi496
提出日時 2021-03-14 18:20:44
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 258 ms / 2,000 ms
コード長 1,011 bytes
コンパイル時間 1,815 ms
コンパイル使用メモリ 178,060 KB
実行使用メモリ 34,816 KB
最終ジャッジ日時 2024-11-06 07:47:43
合計ジャッジ時間 6,074 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
constexpr ll mod=1000000007;
struct UnionFind{
    vector<int> par;
    vector<ll> ans;
    vector<set<int>> st;
    UnionFind(int n):par(n,-1),ans(n,1),st(n){
        for(int i=0;i<n;i++) st[i].insert(i);
    }
    int find(int x){
        return par[x]<0 ? x : par[x]=find(par[x]);
    }
    void unite(int x,int y){
        x=find(x); y=find(y);
        if(x==y)return;
        if(par[x]>par[y]) swap(x,y);
        if(par[x]==par[y]){
            for(int i:st[x]) ans[i]=ans[i]*500000004%mod;
            for(int i:st[y]) ans[i]=ans[i]*500000004%mod;
        }else{
            for(int i:st[y]) ans[i]=0;
        }
        for(int i:st[y]) st[x].insert(i);
        par[x]+=par[y]; par[y]=x;
    }
    int get(int x){
        return ans[x];
    }
};
int main(){
    int N,M; cin>>N>>M;
    UnionFind UF(N);
    for(int i=0;i<M;i++){
        int A,B; cin>>A>>B;
        UF.unite(A-1, B-1);
    }
    for(int i=0;i<N;i++) cout<<UF.get(i)<<endl;
}
0