結果

問題 No.2951 Similar to Mex
ユーザー a1048576a1048576
提出日時 2024-10-25 23:29:13
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,199 bytes
コンパイル時間 5,548 ms
コンパイル使用メモリ 314,916 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-10-25 23:29:52
合計ジャッジ時間 8,268 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,820 KB
testcase_02 AC 2 ms
6,816 KB
testcase_03 AC 2 ms
6,820 KB
testcase_04 AC 2 ms
6,820 KB
testcase_05 AC 2 ms
6,824 KB
testcase_06 WA -
testcase_07 WA -
testcase_08 AC 2 ms
6,820 KB
testcase_09 WA -
testcase_10 AC 2 ms
6,820 KB
testcase_11 AC 13 ms
6,820 KB
testcase_12 WA -
testcase_13 WA -
testcase_14 AC 40 ms
6,820 KB
testcase_15 WA -
testcase_16 AC 35 ms
6,820 KB
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 AC 39 ms
6,816 KB
testcase_22 AC 47 ms
6,816 KB
testcase_23 AC 45 ms
6,816 KB
testcase_24 AC 35 ms
6,820 KB
testcase_25 AC 66 ms
6,820 KB
testcase_26 AC 33 ms
6,820 KB
testcase_27 AC 86 ms
6,816 KB
testcase_28 AC 81 ms
6,820 KB
testcase_29 WA -
testcase_30 AC 83 ms
6,820 KB
testcase_31 AC 82 ms
6,820 KB
testcase_32 AC 83 ms
6,820 KB
testcase_33 WA -
testcase_34 AC 82 ms
6,820 KB
testcase_35 AC 86 ms
6,820 KB
testcase_36 AC 1 ms
6,820 KB
testcase_37 AC 2 ms
6,820 KB
testcase_38 AC 2 ms
6,820 KB
testcase_39 AC 2 ms
6,820 KB
testcase_40 AC 19 ms
6,816 KB
testcase_41 WA -
testcase_42 AC 86 ms
6,816 KB
testcase_43 WA -
testcase_44 AC 19 ms
6,816 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#include <atcoder/all> 
using namespace atcoder;
using mint = modint998244353;
#define int long long
vector<mint> kai,inv;
mint f(int n, int k) {
  if(n < k) return 0;
  return kai[n]*inv[k]*inv[n-k];
}
signed main() {
  int n,m,s;
  cin >> n >> m >> s;
  kai.resize(n+1);
  inv.resize(n+1);
  kai[0] = 1, inv[0] = 1;
  for(int i = 1; i <= n; i++) kai[i] = kai[i-1]*i, inv[i] = inv[i-1]/mint(i);
  mint ans = 0;
  vector<vector<mint>> dp(m+1,vector<mint>(n+1,0));
  dp[0][0] = 1;
  for(int i = 0; i < m; i++) {
    for(int j = 0; j <= n; j++) {
      for(int k = j+1; k <= n; k++) dp[i+1][k]+=dp[i][j]*f(n-j,k-j);
    }
  }
  vector<vector<mint>> dp2(m+2,vector<mint>(m+2,0));
  dp2[0][0] = 1;
  vector<vector<mint>> po(m+2,vector<mint>(m+2,1));
  for(int i = 1; i <= m+1; i++) {
    for(int j = 1; j <= m+1; j++) po[i][j] = po[i][j-1]*i;
  }
  for(int i = 0; i <= m; i++) {
    for(int j = 0; j <= i; j++) {
      for(int k = i+1; k <= m+1; k++) {
        int d = min(s,k)-i;
        if(d < 0) d = 0;
        dp2[k][j+1]+=dp2[i][j]*po[k][d];
      }
    }
  }
  for(int i = 1; i <= m; i++) ans+=dp2[m+1][i]*dp[m-(i-1)][n];
  cout << ans.val() << endl;
}
0