結果

問題 No.2711 Connecting Lights
ユーザー reirei
提出日時 2024-03-31 14:27:43
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 10 ms / 5,000 ms
コード長 1,880 bytes
コンパイル時間 1,128 ms
コンパイル使用メモリ 103,288 KB
実行使用メモリ 6,676 KB
最終ジャッジ日時 2024-03-31 14:27:45
合計ジャッジ時間 1,932 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,676 KB
testcase_01 AC 2 ms
6,676 KB
testcase_02 AC 2 ms
6,676 KB
testcase_03 AC 1 ms
6,676 KB
testcase_04 AC 1 ms
6,676 KB
testcase_05 AC 4 ms
6,676 KB
testcase_06 AC 2 ms
6,676 KB
testcase_07 AC 2 ms
6,676 KB
testcase_08 AC 2 ms
6,676 KB
testcase_09 AC 7 ms
6,676 KB
testcase_10 AC 3 ms
6,676 KB
testcase_11 AC 4 ms
6,676 KB
testcase_12 AC 4 ms
6,676 KB
testcase_13 AC 4 ms
6,676 KB
testcase_14 AC 2 ms
6,676 KB
testcase_15 AC 4 ms
6,676 KB
testcase_16 AC 3 ms
6,676 KB
testcase_17 AC 3 ms
6,676 KB
testcase_18 AC 3 ms
6,676 KB
testcase_19 AC 4 ms
6,676 KB
testcase_20 AC 5 ms
6,676 KB
testcase_21 AC 3 ms
6,676 KB
testcase_22 AC 2 ms
6,676 KB
testcase_23 AC 4 ms
6,676 KB
testcase_24 AC 4 ms
6,676 KB
testcase_25 AC 5 ms
6,676 KB
testcase_26 AC 9 ms
6,676 KB
testcase_27 AC 5 ms
6,676 KB
testcase_28 AC 10 ms
6,676 KB
testcase_29 AC 4 ms
6,676 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <cassert>
#include <cmath>
#include <cstring>
#include <iostream>
#include <set>
#include <tuple>
#include <utility>
#include <vector>
using namespace std;
using i8 = int8_t;
using i32 = int;
using i64 = long long;
using i128 = __int128;
using u64 = unsigned long long;
using ll = long long;

void _main();
int main() {
  cin.tie(0);
  ios::sync_with_stdio(false);
  _main();
}

i64 pow(i64 x, i64 n) {
  i64 res = 1;
  while (n > 0) {
    if (n & 1)
      res *= x;
    x *= x;
    n >>= 1;
  }
  return res;
}

i64 pow(i64 x, i64 n, i64 m) {
  i64 res = 1;
  while (n > 0) {
    if (n & 1)
      res = res * x % m;
    x = x * x % m;
    n >>= 1;
  }
  return res;
}

const i64 inf = 1ll << 60;

void _main() {
  i64 n, m, k;
  cin >> n >> m >> k;
  vector<vector<i64>> dp(m, vector<i64>(n + 1, 0));
  vector<vector<i64>> c(n + 1, vector<i64>(n + 1, 0));
  for (i64 i = 0; i <= n; i++) {
    for (i64 j = 0; j <= n; j++) {
      if (i < j) {
        c[i][j] = 0;
      } else {
        c[i][j] = 1;
        for (i64 a = i; a > 0; a--) {
          c[i][j] *= a;
        }
        for (i64 a = j; a > 0; a--) {
          c[i][j] /= a;
        }
        for (i64 a = i - j; a > 0; a--) {
          c[i][j] /= a;
        }
      }
    }
  }
  for (i64 i = k; i <= n; i++) {
    dp[0][i] = c[n][i];
  }
  for (i64 i = 1; i < m; i++) {
    for (i64 j = k; j <= n; j++) {
      for (i64 a = k; a <= n; a++) {
        for (i64 b = k; b <= n; b++) {
          if (min(j, a) < b) {
            continue;
          }
          dp[i][j] += dp[i - 1][a] * c[a][b] * c[n - a][j - b];
          dp[i][j] %= 998244353;
        }
      }
    }
  }
  i64 ans = 0;
  for (i64 i = k; i <= n; i++) {
    ans = (ans + dp[m - 1][i]) % 998244353;
  }
  if (m == 1) {
    ans = 0;
    for (i64 i = 0; i <= n; i++) {
      ans += c[n][i];
    }
  }
  cout << ans << "\n";
}
0