結果

問題 No.140 みんなで旅行
ユーザー ir5ir5
提出日時 2024-06-16 03:01:25
言語 C++23(gcc13)
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 605 ms / 5,000 ms
コード長 3,194 bytes
コンパイル時間 2,968 ms
コンパイル使用メモリ 147,568 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-16 03:01:32
合計ジャッジ時間 6,705 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,944 KB
testcase_02 AC 8 ms
6,940 KB
testcase_03 AC 2 ms
6,940 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 2 ms
6,940 KB
testcase_07 AC 2 ms
6,944 KB
testcase_08 AC 2 ms
6,940 KB
testcase_09 AC 2 ms
6,944 KB
testcase_10 AC 2 ms
6,940 KB
testcase_11 AC 605 ms
6,940 KB
testcase_12 AC 6 ms
6,940 KB
testcase_13 AC 3 ms
6,944 KB
testcase_14 AC 577 ms
6,940 KB
testcase_15 AC 588 ms
6,944 KB
testcase_16 AC 142 ms
6,940 KB
testcase_17 AC 47 ms
6,940 KB
testcase_18 AC 391 ms
6,944 KB
testcase_19 AC 471 ms
6,940 KB
testcase_20 AC 34 ms
6,940 KB
testcase_21 AC 2 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <cassert>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <iostream>
#include <tuple>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <ranges>
#include <iomanip>
using namespace std;
using ll = long long;

auto range(int n) { return views::iota(0, n); }

template<class T, class U> ostream& operator<<(ostream& os, const pair<T, U>& p){ return os << "{" << p.first << ", " << p.second << "}"; }
template<typename T> ostream& operator<<(ostream& os, const vector<T>& obj) { os << "{"; for (const auto& e : obj) os << e << ", "; return os << "}"; }
template<typename T> ostream& operator<<(ostream& os, const set<T>& obj) { os << "set{"; for (const auto& e : obj) os << e << ", "; return os << "}"; }
template<typename T, typename U> ostream& operator<<(ostream& os, const map<T, U>& obj) { os << "map{"; for (const auto& e : obj) os << e << ", "; return os << "}"; }

#ifdef ONLINE_JUDGE
#define dump(expr) ;
#else
#define dump(expr) { cerr << "\033[33m#L" << __LINE__ << ": " << expr << "\033[39m" << endl; }
#endif

const int mod = 1000000007;
struct mint {
  ll x;
  mint(ll x_ = 0) : x((x_ % mod + mod) % mod) {}
  mint operator-() const { return mint(-x); }
  mint &operator+=(const mint &a) { if ((x += a.x) >= mod) x -= mod; return *this; }
  mint &operator-=(const mint &a) { if ((x += mod - a.x) >= mod) x -= mod; return *this; }
  mint &operator*=(const mint &a) { (x *= a.x) %= mod; return *this; }
  mint operator+(const mint &a) const { mint res(*this); return res += a; }
  mint operator-(const mint &a) const { mint res(*this); return res -= a; }
  mint operator*(const mint &a) const { mint res(*this); return res *= a; }
  mint pow(ll t) const { if (!t) return 1; mint a = pow(t >> 1); a *= a; if (t & 1) a *= *this; return a; }
  mint inv() const { return pow(mod - 2); }
  mint &operator/=(const mint &a) { return (*this) *= a.inv(); }
  mint operator/(const mint &a) const { mint res(*this); return res /= a; }
  friend ostream &operator<<(ostream &os, const mint &m) { os << m.x; return os; }
  friend istream &operator>>(istream &is, mint &m) { is >> m.x; return is; }
};

vector<mint> fact, invfact;
void prepare(int n) {
  fact.resize(n + 1);
  invfact.resize(n + 1);
  fact[0] = 1;
  for (int i : range(n)) fact[i + 1] = fact[i] * (i + 1);
  for (int i : range(n + 1)) invfact[i] = fact[i].inv();
}

mint combin(int n, int r) {
  if (r > n) return mint(0);
  return fact[n] * invfact[r] * invfact[n - r];
}

ll solve() {
  int N; cin >> N;
  prepare(N);

  vector<vector<mint>> dp(N + 1, vector<mint>(N + 1));
  dp[0][0] = 1;
  for (int g : views::iota(1, N + 1))
  for (int n : views::iota(1, N + 1))
  for (int k : range(n)) {
    dp[g][n] += combin(n - 1, k) * dp[g - 1][n - 1 - k];
  }
  dump(dp);

  mint res = 0;
  for (int g : range(N + 1))
  for (int n : range(N + 1)) {
    if (dp[g][n].x) {
      dump(g << " " << n << " => " << dp[g][n] << " " << (mint(g) * (g - 1)).pow(N - n));
    }
    res += dp[g][n] * combin(N, n) * (mint(g) * (g - 1)).pow(N - n);
  }
  dump(res);
  return res.x;
}

int main() {
  cout << fixed << setprecision(12);
  cout << solve() << endl;
}
0