結果

問題 No.1451 集団登校
ユーザー tnakao0123tnakao0123
提出日時 2021-04-01 15:28:04
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 53 ms / 2,000 ms
コード長 3,316 bytes
コンパイル時間 963 ms
コンパイル使用メモリ 95,564 KB
実行使用メモリ 15,940 KB
最終ジャッジ日時 2023-08-22 18:30:56
合計ジャッジ時間 4,009 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
11,816 KB
testcase_01 AC 3 ms
11,696 KB
testcase_02 AC 3 ms
11,632 KB
testcase_03 AC 3 ms
11,624 KB
testcase_04 AC 3 ms
11,716 KB
testcase_05 AC 3 ms
11,776 KB
testcase_06 AC 4 ms
11,788 KB
testcase_07 AC 3 ms
11,692 KB
testcase_08 AC 35 ms
15,496 KB
testcase_09 AC 4 ms
11,676 KB
testcase_10 AC 22 ms
14,152 KB
testcase_11 AC 12 ms
13,568 KB
testcase_12 AC 22 ms
13,992 KB
testcase_13 AC 42 ms
15,412 KB
testcase_14 AC 52 ms
15,940 KB
testcase_15 AC 25 ms
13,908 KB
testcase_16 AC 38 ms
15,116 KB
testcase_17 AC 19 ms
11,844 KB
testcase_18 AC 16 ms
13,148 KB
testcase_19 AC 9 ms
12,912 KB
testcase_20 AC 53 ms
15,920 KB
testcase_21 AC 13 ms
11,724 KB
testcase_22 AC 12 ms
12,356 KB
testcase_23 AC 6 ms
11,876 KB
testcase_24 AC 20 ms
14,008 KB
testcase_25 AC 25 ms
14,028 KB
testcase_26 AC 37 ms
15,352 KB
testcase_27 AC 37 ms
13,988 KB
testcase_28 AC 14 ms
12,752 KB
testcase_29 AC 27 ms
13,896 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

/* -*- coding: utf-8 -*-
 *
 * 1451.cc:  No.1451 集団登校 - yukicoder
 */

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<stack>
#include<list>
#include<queue>
#include<deque>
#include<algorithm>
#include<numeric>
#include<utility>
#include<complex>
#include<functional>
 
using namespace std;

/* constant */

const int MAX_N = 100000;
const int MAX_M = 100000;
const int MAX_GN = MAX_N + MAX_M;
const int MOD = 1000000007;

/* typedef */

typedef long long ll;
typedef vector<int> vi;

struct UFT {
  int links[MAX_N], ranks[MAX_N], sizes[MAX_N];
  UFT() {}

  void init(int n) {
    for (int i = 0; i < n; i++) links[i] = i, ranks[i] = sizes[i] = 1;
  }

  int root(int i) {
    int i0 = i;
    while (links[i0] != i0) i0 = links[i0];
    return (links[i] = i0);
  }

  int rank(int i) { return ranks[root(i)]; }
  int size(int i) { return sizes[root(i)]; }
  bool same(int i, int j) { return root(i) == root(j); }

  int merge(int i0, int i1) {
    int r0 = root(i0), r1 = root(i1), mr;
    if (r0 == r1) return r0;
    if (ranks[r0] == ranks[r1]) {
      links[r1] = r0;
      sizes[r0] += sizes[r1];
      ranks[r0]++;
      mr = r0;
    }
    else if (ranks[r0] > ranks[r1]) {
      links[r1] = r0;
      sizes[r0] += sizes[r1];
      mr = r0;
    }
    else {
      links[r0] = r1;
      sizes[r1] += sizes[r0];
      mr = r1;
    }
    return mr;
  }
};

/* global variables */

UFT uft;
int grts[MAX_N], gszs[MAX_GN], ps[MAX_GN];
vi nbrs[MAX_GN];
int ls[MAX_GN], rs[MAX_GN], cis[MAX_GN];
int zs[MAX_N + 1], es[MAX_N + 1];

/* subroutines */

int powmod(int a, int n) {  // a^n % MOD
  int pm = 1;
  while (n > 0) {
    if (n & 1) pm = (ll)pm * a % MOD;
    a = (ll)a * a % MOD;
    n >>= 1;
  }
  return pm;
}

/* main */

int main() {
  int n, m;
  scanf("%d%d", &n, &m);

  uft.init(n);
  for (int i = 0; i < n; i++) grts[i] = i, gszs[i] = 1;

  int gn = n;
  memset(ps, -1, sizeof(ps));

  for (int i = 0; i < m; i++) {
    int a, b;
    scanf("%d%d", &a, &b);
    a--, b--;

    int ra = uft.root(a), rb = uft.root(b);
    if (ra != rb) {
      int ga = grts[ra], gb = grts[rb];
      gszs[gn] = gszs[ga] + gszs[gb];
      ps[ga] = ps[gb] = gn;
      nbrs[gn].push_back(ga);
      nbrs[gn].push_back(gb);

      int r = uft.merge(ra, rb);
      grts[r] = gn++;
    }
  }

  for (int st = 0, k = 0; st < gn; st++)
    if (ps[st] < 0) {
      for (int u = st; u >= 0;) {
	vi &nbru = nbrs[u];
	if (cis[u] == 0) {
	  ls[u] = k;
	  if (u < n) k++;
	}
	if (cis[u] < nbru.size()) {
	  int v = nbru[cis[u]++];
	  u = v;
	}
	else {
	  rs[u] = k;
	  if (! nbru.empty()) {
	    int v0 = nbru[0], v1 = nbru[1];
	    if (gszs[v0] < gszs[v1])
	      zs[ls[v0]]++, zs[rs[v0]]--;
	    else if (gszs[v0] > gszs[v1])
	      zs[ls[v1]]++, zs[rs[v1]]--;
	    else
	      es[ls[u]]++, es[rs[u]]--;
	  }
	  u = ps[u];
	}
      }
    }
  //for (int i = 0; i < gn; i++) printf("%d: %d-%d\n", i, ls[i], rs[i]);

  for (int i = 0; i < n; i++)
    zs[i + 1] += zs[i], es[i + 1] += es[i];

  int inv2 = powmod(2, MOD - 2);
  for (int i = 0; i < n; i++) {
    int v = (zs[ls[i]] > 0) ? 0 : powmod(inv2, es[ls[i]]);
    printf("%d\n", v);
    //printf("%d: z=%d, e=%d\n", i, zs[ls[i]], es[ls[i]]);
  }
  return 0;
}
0