結果

問題 No.1451 集団登校
コンテスト
ユーザー こばゆ
提出日時 2021-03-31 15:45:13
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,125 bytes
コンパイル時間 901 ms
コンパイル使用メモリ 75,988 KB
実行使用メモリ 10,624 KB
最終ジャッジ日時 2024-12-15 01:08:00
合計ジャッジ時間 49,120 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 3
other WA * 12 TLE * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
using namespace std;

// Union-Find
struct UnionFind {
    vector<int> par, siz;

    // 初期化
    UnionFind(int n) : par(n, -1) , siz(n, 1) { }

    // 根を求める
    int root(int x) {
        if (par[x] == -1) return x; // x が根の場合は x を返す
        else return par[x] = root(par[x]);
    }

    // x と y が同じグループに属するかどうか (根が一致するかどうか)
    bool issame(int x, int y) {
        return root(x) == root(y);
    }

    // x を含むグループと y を含むグループとを併合する
    bool unite(int x, int y) {
        // x, y をそれぞれ根まで移動する
        x = root(x);
        y = root(y);

        // すでに同じグループのときは何もしない
        if (x == y) return false;

        // union by size (y 側のサイズが小さくなるようにする)
        if (siz[x] < siz[y]) swap(x, y);

        // y を x の子とする
        par[y] = x;
        siz[x] += siz[y];
        return true;
    }

    // x を含むグループのサイズ
    int size(int x) {
        return siz[root(x)];
    }
};
int main(int argc, char const *argv[]) {
  int n,m;
  std::cin >> n>>m;
  std::vector<int> x(n,0);
  std::vector<bool> y(n,true);
  UnionFind uf(n);
  for (size_t i = 0; i < m; i++) {
    int a,b;
    std::cin >> a>>b;
    a--;
    b--;
    if (uf.size(a)>uf.size(b)) {
      y.at(uf.root(b))=false;
      uf.unite(a,b);
    } else if (uf.size(a)<uf.size(b)) {
      y.at(uf.root(a))=false;
      uf.unite(a,b);
    } else {
      uf.unite(a,b);
      for (size_t j = 0; j < n; j++) {
        if (uf.issame(a,j)) {
          x.at(j)+=2;
        }
      }
    }
  }
  for (size_t i = 0; i < n; i++) {
    if (y.at(i)) {
      if (x.at(i)==1) {
        std::cout << 1 << '\n';
      } else {
        int ans=1000000008;
        for (size_t j = 0; j < x.at(i); j++) {
          if (ans%2) {
            ans+=1000000008;
          }
          ans/=2;
        }
        std::cout << ans << '\n';
      }
    } else {
      std::cout << 0 << '\n';
    }
  }
  return 0;
}
0