結果

問題 No.827 総神童数
コンテスト
ユーザー 0w1
提出日時 2020-11-14 22:56:31
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 134 ms / 2,000 ms
コード長 2,398 bytes
コンパイル時間 2,241 ms
コンパイル使用メモリ 201,740 KB
最終ジャッジ日時 2025-01-16 00:42:20
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

template<int m>
struct Mint {
  int v;
  Mint(int _v = 0) : v(_v) { v %= m; adjust(); }
  void adjust() { if (v < 0) v += m; if (v >= m) v -= m; assert(0 <= v && v < m); }
  Mint operator+() const { return *this; }
  Mint operator-() const { Mint r = Mint(0) - *this; return r.adjust(), r; }
  friend Mint operator+(Mint a, Mint b) { a.v += b.v; return a.adjust(), a; }
  Mint& operator+=(const Mint &b) { return v += b.v, adjust(), *this; }
  friend Mint operator-(Mint a, Mint b) { a.v -= b.v; return a.adjust(), a; }
  Mint& operator-=(const Mint &b) { return v -= b.v, adjust(), *this; }
  friend Mint operator*(Mint a, Mint b) { a.v = (int64_t) a.v * b.v % m; return a.adjust(), a; }
  friend Mint operator*(Mint a, int v) { return a * Mint(v); }
  friend Mint operator^(Mint a, int p) {
    int v = a.v;
    int r = 1;
    for (int i = p; i; i >>= 1) {
      if (i & 1) r = (int64_t) r * v % m;
      v = (int64_t) v * v % m;
    }
    return r;
  }
  friend Mint operator^(Mint a, Mint p) { return a^p.v; }
  friend Mint operator/(Mint a, Mint b) { return a * (Mint(b.v)^(m-2)); }
  friend ostream& operator<<(ostream &os, Mint a) { return os << a.v, os; }
};

template<int m, int MAXN>
struct Factorial {
  vector<Mint<m>> fac, ifac;
  Factorial() : fac(MAXN + 1), ifac(MAXN + 1) {
    fac[0] = Mint<m>(1);
    for (int i = 1; i <= MAXN; ++i) fac[i] = fac[i - 1] * i;
    ifac[MAXN] = Mint<m>(1) / fac[MAXN];
    for (int i = MAXN; i; --i) ifac[i - 1] = ifac[i] * i;
  }
  Mint<m> factorial(int n) { return fac[n]; }
  Mint<m> binomial(int a, int b) { return fac[a] * ifac[b] * ifac[a - b]; };
};


int main() {
  ios::sync_with_stdio(false);
  
  int N; { cin >> N; }

  vector<vector<int>> G(N); {
    for (int i = 0; i + 1 < N; ++i) {
      int U, V; {
        cin >> U >> V;
        --U, --V;
      }
      G[U].push_back(V);
      G[V].push_back(U);
    }
  }

  const int M = 1e9 + 7;

  Mint<M> res; {
    vector<int> dcnt(N); {
      function<void(int, int, int)> dfs = [&](int u, int fa, int d) {
        dcnt[d] += 1;
        for (int v : G[u]) if (v != fa) {
          dfs(v, u, d + 1);
        }
      };
      dfs(0, -1, 0);
    }

    Factorial<M, 1 << 18> ft;

    for (int d = 0; d < N; ++d) {
      res = res + ft.binomial(N, d + 1) * ft.factorial(d) * ft.factorial(N - (d + 1)) * dcnt[d];
    }
  }

  cout << res << endl;
}
0