結果

問題 No.827 総神童数
ユーザー 0w10w1
提出日時 2020-11-14 22:56:31
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 156 ms / 2,000 ms
コード長 2,398 bytes
コンパイル時間 2,598 ms
コンパイル使用メモリ 210,448 KB
実行使用メモリ 31,172 KB
最終ジャッジ日時 2024-07-23 00:54:57
合計ジャッジ時間 8,504 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 8 ms
6,812 KB
testcase_01 AC 7 ms
6,944 KB
testcase_02 AC 7 ms
6,944 KB
testcase_03 AC 8 ms
6,940 KB
testcase_04 AC 8 ms
6,944 KB
testcase_05 AC 8 ms
6,944 KB
testcase_06 AC 8 ms
6,944 KB
testcase_07 AC 8 ms
6,944 KB
testcase_08 AC 8 ms
6,944 KB
testcase_09 AC 127 ms
31,172 KB
testcase_10 AC 43 ms
13,184 KB
testcase_11 AC 9 ms
6,940 KB
testcase_12 AC 22 ms
9,344 KB
testcase_13 AC 100 ms
26,740 KB
testcase_14 AC 10 ms
6,944 KB
testcase_15 AC 40 ms
13,696 KB
testcase_16 AC 102 ms
21,336 KB
testcase_17 AC 120 ms
25,296 KB
testcase_18 AC 53 ms
15,616 KB
testcase_19 AC 156 ms
16,700 KB
testcase_20 AC 99 ms
13,696 KB
testcase_21 AC 66 ms
11,520 KB
testcase_22 AC 107 ms
14,460 KB
testcase_23 AC 8 ms
6,940 KB
testcase_24 AC 129 ms
15,448 KB
testcase_25 AC 94 ms
13,028 KB
testcase_26 AC 125 ms
15,492 KB
testcase_27 AC 79 ms
12,160 KB
testcase_28 AC 75 ms
12,032 KB
testcase_29 AC 58 ms
10,672 KB
testcase_30 AC 20 ms
6,944 KB
testcase_31 AC 43 ms
9,344 KB
testcase_32 AC 42 ms
9,216 KB
testcase_33 AC 143 ms
16,352 KB
testcase_34 AC 117 ms
15,392 KB
testcase_35 AC 45 ms
9,600 KB
testcase_36 AC 65 ms
11,392 KB
testcase_37 AC 91 ms
12,984 KB
testcase_38 AC 133 ms
16,192 KB
権限があれば一括ダウンロードができます

ソースコード

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