Problem p02728 (C++) — buggy submission, verdict: Runtime Error

#include <algorithm>
#include <bitset>
#include <cassert>
#include <climits>
#include <cmath>
#include <complex>
#include <cstdio>
#include <cstring>
#include <ctime>
#include <deque>
#include <functional>
#include <iostream>
#include <limits>
#include <list>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <utility>
#include <vector>

using namespace std;
using lint = long long int;
long long int INF = 1001001001001001LL;
int inf = 1000000007;
long long int MOD = 1000000007LL;
double PI = 3.1415926535897932;

template <typename T1, typename T2> inline void chmin(T1 &a, const T2 &b) {
  if (a > b)
    a = b;
}
template <typename T1, typename T2> inline void chmax(T1 &a, const T2 &b) {
  if (a < b)
    a = b;
}

#define ALL(a) a.begin(), a.end()
#define RALL(a) a.rbegin(), a.rend()

/* do your best */

const long long MAXN = 1001024;

// a, b の最大公約数を返す O( log max(a, b) )
long long gcd(long long a, long long b) {
  if (b == 0)
    return a;
  return gcd(b, a % b);
}

// ax + by = 1 となるような (x, y) と gcd(a, b) を返す. gcd(a, b) = 1
// の時、解が存在する.
long long extgcd(long long a, long long b, long long &x, long long &y) {
  long long d = a;
  if (b != 0) {
    d = extgcd(b, a, y, x);
    y -= (a / b) * x;
  } else {
    x = 1;
    y = 0;
  }
  return d;
}

// 区間 [a, b) に存在する素数の個数を返す関数

long long prime[MAXN];  // [a,b) の素数のうち i 番目の素数
bool is_prime[MAXN];    // 整数 i が素数であるかどうか
bool is_prime_ab[MAXN]; // 整数 i+a が素数であるかどうか

long long sieve(long long n) {
  long long res = 0;
  fill(is_prime, is_prime + MAXN, true);
  is_prime[0] = is_prime[1] = false; // 0 と 1 は素数ではない。
  for (long long i = 2; i <= n; ++i) {
    if (!is_prime[i])
      continue;
    prime[res++] = i;
    for (long long j = 2 * i; j <= n; j += i)
      is_prime[j] = false; // 素数 i の倍数は素数ではない (ふるい(篩)にかける)
  }
  return res;
}

/*
long long segment_sieve(long long a, long long b){
  fill(is_prime, is_prime + MAXN, true);
  fill(is_prime_ab, is_prime_ab + MAXN, true);
  for(long long i = 2; i * i <= b - 1; i++) {
    if(!is_prime[i]) continue;
    for(long long j = 2 * i; j * j <= b - 1; j += i) is_prime[j] = false; //
素数 i で篩にかける for(long long j = a - a % i; j < b; j += i) { if(j < a)
continue; if(is_prime_ab[j-a]) is_prime_ab[j-a] = false; // 素数 i で篩にかける
    }
  }

  long long res = 0;
  for(long long i = a; i < b; i++) if(is_prime_ab[i - a]) prime[res++] = i;
  return res;
}
*/

// ある整数の約数列挙
vector<long long> divisors(long long n) {
  vector<long long> res;
  for (long long i = 1; i * i <= n; ++i) {
    if (n % i != 0)
      continue;
    res.push_back(i);
    if (n / i == i)
      continue; // 上の行で追加済み。
    res.push_back(n / i);
  }
  sort(res.begin(), res.end());
  return res;
}

// 素因数分解
map<long long, long long> prime_factors(long long n) {
  map<long long, long long> res;
  if (n == 1) { // n=1 の素因数分解は n^1
    res[n] = 1;
    return res;
  }
  for (long long i = 2, _n = n; i * i <= _n; ++i) {
    while (n % i == 0) {
      ++res[i]; // 素数i^{res[i]}
      n /= i;
    }
  }
  if (n != 1)
    res[n] = 1;
  return res;
}

// 繰り返し 2 乗法
long long modpow(long long a, long long n) {
  long long res = 1;
  while (n > 0) {
    if (n & 1)
      res = res * a % MOD;
    a = a * a % MOD;
    n >>= 1;
  }
  return res;
}

// 逆元を求める. a と m は互いに素であることが要請される.
long long modinv(long long a, long long m) {
  long long b = m, u = 1, v = 0;
  while (b) {
    long long t = a / b;
    a -= t * b;
    swap(a, b);
    u -= t * v;
    swap(u, v);
  }
  u %= m;
  if (u < 0)
    u += m;
  return u;
}

long long fac[MAXN], finv[MAXN], inv[MAXN];

// 前処理 O(n)
void math_init() {
  fac[0] = fac[1] = 1;
  finv[0] = finv[1] = 1;
  inv[1] = 1;
  for (long long i = 2; i < MAXN; i++) {
    fac[i] = fac[i - 1] * i % MOD;
    inv[i] = MOD - inv[MOD % i] * (MOD / i) % MOD;
    finv[i] = finv[i - 1] * inv[i] % MOD;
  }
}

// 二項係数計算 O(1)
long long COM(long long n, long long k) {
  if (n < k)
    return 0;
  if (n < 0 || k < 0)
    return 0;
  return fac[n] * (finv[k] * finv[n - k] % MOD) % MOD;
}

// before your coding, you have to write a line "math_init()"

void dfs1(int cur, int par, vector<vector<int>> &g, vector<lint> &sz,
          vector<lint> &dp1) {
  for (auto nxt : g[cur]) {
    if (nxt == par)
      continue;
    dfs1(nxt, cur, g, sz, dp1);
  }

  int s = 1;
  int ccnt = 0;
  for (auto nxt : g[cur]) {
    if (nxt == par)
      continue;
    s += sz[nxt];
    ccnt += sz[nxt];
  }

  sz[cur] = s;

  if (ccnt == 0) {
    dp1[cur] = 1;
    return;
  }

  lint val = 1;
  for (auto nxt : g[cur]) {
    if (nxt == par)
      continue;
    val *= dp1[nxt];
    val %= MOD;
    val *= COM(ccnt, sz[nxt]);
    val %= MOD;
    ccnt -= sz[nxt];
  }
  dp1[cur] = val;
}

void dfs2(int cur, int par, vector<vector<int>> &g, vector<lint> &sz,
          vector<lint> &dp1, vector<lint> &ans, lint D) {

  // cout << cur + 1 << " : " <<  ", Dpar = " << D << endl;
  int n = g.size();
  if (D == -1) {
    ans[cur] = dp1[cur];
  } else {

    // ans を計算
    // 子供と Dpar の情報を使う
    lint val = D;
    lint cnt = n - 1;
    for (auto nxt : g[cur]) {
      if (nxt == par)
        continue;
      val *= dp1[nxt];
      val %= MOD;
      val *= COM(cnt, sz[nxt]);
      val %= MOD;
      cnt -= sz[nxt];
    }

    ans[cur] = val;
  }

  // Dpar を計算しつつ潜る
  for (auto nxt : g[cur]) {
    if (nxt == par)
      continue;

    // D は，nxt 以外の子供の情報から得られるもの
    lint nxtD = ans[cur];
    nxtD %= MOD;
    nxtD *= modinv(dp1[nxt], MOD);
    nxtD %= MOD;
    nxtD *= finv[n - 1];
    nxtD %= MOD;
    nxtD *= fac[n - 1 - sz[nxt]];
    nxtD %= MOD;
    nxtD *= fac[sz[nxt]];
    nxtD %= MOD;
    dfs2(nxt, cur, g, sz, dp1, ans, nxtD);
  }
}

int main() {

  math_init();

  int n;
  cin >> n;
  vector<vector<int>> g(n);
  for (int i = 0; i < n - 1; i++) {
    int a, b;
    cin >> a >> b;
    a--;
    b--;
    g[a].push_back(b);
    g[b].push_back(a);
  }
  vector<lint> sz(n, -1);
  vector<lint> dp1(n, -1);
  dfs1(0, -1, g, sz, dp1);
  vector<lint> ans(n, -1);
  dfs2(0, -1, g, sz, dp1, ans, -1);

  for (auto x : ans)
    cout << x << endl;
  return 0;
}


Error: 0

Fixed submission:

#include <algorithm>
#include <bitset>
#include <cassert>
#include <climits>
#include <cmath>
#include <complex>
#include <cstdio>
#include <cstring>
#include <ctime>
#include <deque>
#include <functional>
#include <iostream>
#include <limits>
#include <list>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <utility>
#include <vector>

using namespace std;
using lint = long long int;
long long int INF = 1001001001001001LL;
int inf = 1000000007;
long long int MOD = 1000000007LL;
double PI = 3.1415926535897932;

template <typename T1, typename T2> inline void chmin(T1 &a, const T2 &b) {
  if (a > b)
    a = b;
}
template <typename T1, typename T2> inline void chmax(T1 &a, const T2 &b) {
  if (a < b)
    a = b;
}

#define ALL(a) a.begin(), a.end()
#define RALL(a) a.rbegin(), a.rend()

/* do your best */

const long long MAXN = 1001024;

// a, b の最大公約数を返す O( log max(a, b) )
long long gcd(long long a, long long b) {
  if (b == 0)
    return a;
  return gcd(b, a % b);
}

// ax + by = 1 となるような (x, y) と gcd(a, b) を返す. gcd(a, b) = 1
// の時、解が存在する.
long long extgcd(long long a, long long b, long long &x, long long &y) {
  long long d = a;
  if (b != 0) {
    d = extgcd(b, a, y, x);
    y -= (a / b) * x;
  } else {
    x = 1;
    y = 0;
  }
  return d;
}

// 区間 [a, b) に存在する素数の個数を返す関数

long long prime[MAXN];  // [a,b) の素数のうち i 番目の素数
bool is_prime[MAXN];    // 整数 i が素数であるかどうか
bool is_prime_ab[MAXN]; // 整数 i+a が素数であるかどうか

long long sieve(long long n) {
  long long res = 0;
  fill(is_prime, is_prime + MAXN, true);
  is_prime[0] = is_prime[1] = false; // 0 と 1 は素数ではない。
  for (long long i = 2; i <= n; ++i) {
    if (!is_prime[i])
      continue;
    prime[res++] = i;
    for (long long j = 2 * i; j <= n; j += i)
      is_prime[j] = false; // 素数 i の倍数は素数ではない (ふるい(篩)にかける)
  }
  return res;
}

/*
long long segment_sieve(long long a, long long b){
  fill(is_prime, is_prime + MAXN, true);
  fill(is_prime_ab, is_prime_ab + MAXN, true);
  for(long long i = 2; i * i <= b - 1; i++) {
    if(!is_prime[i]) continue;
    for(long long j = 2 * i; j * j <= b - 1; j += i) is_prime[j] = false; //
素数 i で篩にかける for(long long j = a - a % i; j < b; j += i) { if(j < a)
continue; if(is_prime_ab[j-a]) is_prime_ab[j-a] = false; // 素数 i で篩にかける
    }
  }

  long long res = 0;
  for(long long i = a; i < b; i++) if(is_prime_ab[i - a]) prime[res++] = i;
  return res;
}
*/

// ある整数の約数列挙
vector<long long> divisors(long long n) {
  vector<long long> res;
  for (long long i = 1; i * i <= n; ++i) {
    if (n % i != 0)
      continue;
    res.push_back(i);
    if (n / i == i)
      continue; // 上の行で追加済み。
    res.push_back(n / i);
  }
  sort(res.begin(), res.end());
  return res;
}

// 素因数分解
map<long long, long long> prime_factors(long long n) {
  map<long long, long long> res;
  if (n == 1) { // n=1 の素因数分解は n^1
    res[n] = 1;
    return res;
  }
  for (long long i = 2, _n = n; i * i <= _n; ++i) {
    while (n % i == 0) {
      ++res[i]; // 素数i^{res[i]}
      n /= i;
    }
  }
  if (n != 1)
    res[n] = 1;
  return res;
}

// 繰り返し 2 乗法
long long modpow(long long a, long long n) {
  long long res = 1;
  while (n > 0) {
    if (n & 1)
      res = res * a % MOD;
    a = a * a % MOD;
    n >>= 1;
  }
  return res;
}

// 逆元を求める. a と m は互いに素であることが要請される.
long long modinv(long long a, long long m) {
  long long b = m, u = 1, v = 0;
  while (b) {
    long long t = a / b;
    a -= t * b;
    swap(a, b);
    u -= t * v;
    swap(u, v);
  }
  u %= m;
  if (u < 0)
    u += m;
  return u;
}

long long fac[MAXN], finv[MAXN], inv[MAXN];

// 前処理 O(n)
void math_init() {
  fac[0] = fac[1] = 1;
  finv[0] = finv[1] = 1;
  inv[1] = 1;
  for (long long i = 2; i < MAXN; i++) {
    fac[i] = fac[i - 1] * i % MOD;
    inv[i] = MOD - inv[MOD % i] * (MOD / i) % MOD;
    finv[i] = finv[i - 1] * inv[i] % MOD;
  }
}

// 二項係数計算 O(1)
long long COM(long long n, long long k) {
  if (n < k)
    return 0;
  if (n < 0 || k < 0)
    return 0;
  return fac[n] * (finv[k] * finv[n - k] % MOD) % MOD;
}

// before your coding, you have to write a line "math_init()"

void dfs1(int cur, int par, vector<vector<int>> &g, vector<lint> &sz,
          vector<lint> &dp1) {
  for (auto nxt : g[cur]) {
    if (nxt == par)
      continue;
    dfs1(nxt, cur, g, sz, dp1);
  }

  int s = 1;
  int ccnt = 0;
  for (auto nxt : g[cur]) {
    if (nxt == par)
      continue;
    s += sz[nxt];
    ccnt += sz[nxt];
  }

  sz[cur] = s;

  if (ccnt == 0) {
    dp1[cur] = 1;
    return;
  }

  lint val = 1;
  for (auto nxt : g[cur]) {
    if (nxt == par)
      continue;
    val *= dp1[nxt];
    val %= MOD;
    val *= COM(ccnt, sz[nxt]);
    val %= MOD;
    ccnt -= sz[nxt];
  }
  dp1[cur] = val;
}

void dfs2(int cur, int par, vector<vector<int>> &g, vector<lint> &sz,
          vector<lint> &dp1, vector<lint> &ans, lint D) {

  // cout << cur + 1 << " : " <<  ", Dpar = " << D << endl;
  int n = g.size();
  if (D == -1) {
    ans[cur] = dp1[cur];
  } else {

    // ans を計算
    // 子供と Dpar の情報を使う
    lint val = D;
    lint cnt = n - 1;
    for (auto nxt : g[cur]) {
      if (nxt == par)
        continue;
      val *= dp1[nxt];
      val %= MOD;
      val *= COM(cnt, sz[nxt]);
      val %= MOD;
      cnt -= sz[nxt];
    }

    ans[cur] = val;
  }

  // Dpar を計算しつつ潜る
  for (auto nxt : g[cur]) {
    if (nxt == par)
      continue;

    // D は，nxt 以外の子供の情報から得られるもの
    lint nxtD = ans[cur];
    nxtD %= MOD;
    nxtD *= inv[dp1[nxt]];
    nxtD %= MOD;
    nxtD *= finv[n - 1];
    nxtD %= MOD;
    nxtD *= fac[n - 1 - sz[nxt]];
    nxtD %= MOD;
    nxtD *= fac[sz[nxt]];
    nxtD %= MOD;
    dfs2(nxt, cur, g, sz, dp1, ans, nxtD);
  }
}

int main() {

  math_init();

  int n;
  cin >> n;
  vector<vector<int>> g(n);
  for (int i = 0; i < n - 1; i++) {
    int a, b;
    cin >> a >> b;
    a--;
    b--;
    g[a].push_back(b);
    g[b].push_back(a);
  }
  vector<lint> sz(n, -1);
  vector<lint> dp1(n, -1);
  dfs1(0, -1, g, sz, dp1);
  vector<lint> ans(n, -1);
  dfs2(0, -1, g, sz, dp1, ans, -1);

  for (auto x : ans)
    cout << x << endl;
  return 0;
}