結果

問題 No.1451 集団登校
ユーザー hiro71687k
提出日時 2023-07-18 14:33:15
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 151 ms / 2,000 ms
コード長 2,663 bytes
コンパイル時間 3,690 ms
コンパイル使用メモリ 254,096 KB
最終ジャッジ日時 2025-02-15 15:34:00
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace atcoder;
using namespace std;
using ll=long long;
using ld=long double;
ld pie=3.141592653589793;
ll mod=1000000007;
ld inf=10000999999999900;
struct unionfind
{
    vector<ll>par,siz;
    unionfind(ll n): par(n,-1),siz(n,1){}
    ll root(ll x){
        if (par[x]==-1)
        {
            return x;
        }else{
            return par[x]=root(par[x]);
        }
    }
    bool issame(ll x,ll y){
        return root(x)==root(y);
    }
    bool unite(ll x,ll y){
        x=root(x);
        y=root(y);
        if (x==y)
        {
            return false;
        }
        if (siz[x]<siz[y])
        {
            swap(x,y);
        }
        par[y]=x;
        siz[x]+=siz[y];
        return true;
    }
    ll size(ll x){
        return siz[root(x)];
    }
};
int main(){
    ll n,m;
    cin >> n >> m;
    vector<ll>a(m),b(m);
    unionfind uf(n);
    for (ll i = 0; i < m; i++)
    {
        cin >> a[i]  >> b[i];
        a[i]--,b[i]--;
    }
    vector<ll>ans(n,1);
    vector<ll>g(n);
    for (ll i = 0; i < n; i++)
    {
        g[i]=i;
    }
    for (ll i = 0; i < m; i++)
    {
        if (uf.issame(a[i],b[i]))
        {
            continue;
        }
        
        ll aa=uf.root(a[i]);
        ll bb=uf.root(b[i]);
        ll x=uf.size(aa);
        ll y=uf.size(bb);
        if (x>y)
        {
            uf.unite(aa,bb);
            ans[uf.root(aa)]=ans[aa];
            g[bb]=uf.root(aa);
            g[aa]=uf.root(aa);
            ans[bb]=0;
        }else if (x==y)
        {
            uf.unite(aa,bb);
            ll r=uf.root(aa);
            if (aa==r)
            {
                g[bb]=aa;
                ans[aa]*=inv_mod(2,mod);
                ans[aa]%=mod;
                ans[bb]*=inv_mod(ans[aa],mod);
                ans[bb]%=mod;
                ans[bb]*=inv_mod(2,mod);
                ans[bb]%=mod;
            }else{
                g[aa]=bb;
                ans[bb]*=inv_mod(2,mod);
                ans[bb]%=mod;
                ans[aa]*=inv_mod(ans[bb],mod);
                ans[aa]%=mod;
                ans[aa]*=inv_mod(2,mod);
                ans[aa]%=mod;
            }
        }else{
            uf.unite(aa,bb);
            ans[uf.root(aa)]=ans[bb];
            g[bb]=uf.root(aa);
            g[aa]=uf.root(aa);
            ans[aa]=0;
        }
    }
    for (ll i = 0; i < n; i++)
    {
        ll now=i;
        ll x=ans[now];
        while (1)
        {
            if (now==g[now])
            {
                break;
            }
            now=g[now];
            x*=ans[now];
            x%=mod;
        }
        cout << x << endl;
    }
}
0