結果

問題 No.2951 Similar to Mex
ユーザー pockynypockyny
提出日時 2024-10-26 18:29:54
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 343 ms / 2,000 ms
コード長 1,119 bytes
コンパイル時間 1,128 ms
コンパイル使用メモリ 74,352 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-26 18:30:02
合計ジャッジ時間 8,198 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,820 KB
testcase_01 AC 2 ms
6,816 KB
testcase_02 AC 3 ms
6,820 KB
testcase_03 AC 2 ms
6,816 KB
testcase_04 AC 3 ms
6,816 KB
testcase_05 AC 3 ms
6,816 KB
testcase_06 AC 3 ms
6,820 KB
testcase_07 AC 3 ms
6,816 KB
testcase_08 AC 3 ms
6,816 KB
testcase_09 AC 2 ms
6,820 KB
testcase_10 AC 3 ms
6,820 KB
testcase_11 AC 69 ms
6,820 KB
testcase_12 AC 6 ms
6,820 KB
testcase_13 AC 6 ms
6,816 KB
testcase_14 AC 124 ms
6,816 KB
testcase_15 AC 177 ms
6,820 KB
testcase_16 AC 71 ms
6,816 KB
testcase_17 AC 81 ms
6,820 KB
testcase_18 AC 134 ms
6,816 KB
testcase_19 AC 79 ms
6,816 KB
testcase_20 AC 136 ms
6,816 KB
testcase_21 AC 176 ms
6,816 KB
testcase_22 AC 124 ms
6,816 KB
testcase_23 AC 253 ms
6,820 KB
testcase_24 AC 229 ms
6,820 KB
testcase_25 AC 276 ms
6,816 KB
testcase_26 AC 210 ms
6,820 KB
testcase_27 AC 339 ms
6,820 KB
testcase_28 AC 315 ms
6,820 KB
testcase_29 AC 335 ms
6,816 KB
testcase_30 AC 329 ms
6,820 KB
testcase_31 AC 320 ms
6,816 KB
testcase_32 AC 315 ms
6,816 KB
testcase_33 AC 324 ms
6,820 KB
testcase_34 AC 342 ms
6,816 KB
testcase_35 AC 328 ms
6,820 KB
testcase_36 AC 2 ms
6,816 KB
testcase_37 AC 2 ms
6,816 KB
testcase_38 AC 2 ms
6,816 KB
testcase_39 AC 3 ms
6,820 KB
testcase_40 AC 42 ms
6,820 KB
testcase_41 AC 3 ms
6,816 KB
testcase_42 AC 42 ms
6,816 KB
testcase_43 AC 2 ms
6,816 KB
testcase_44 AC 343 ms
6,816 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <atcoder/modint>

using namespace std;
using namespace atcoder;
using mint = modint998244353;
mint dp[310][310],S[310][310],f[310];
mint pw(mint a,int x){
    mint ret = 1;
    while(x){
        if(x&1) ret *= a;
        a *= a; x /= 2;
    }
    return ret;
}

int main(){
    int i,j,n,m,k; cin >> n >> m >> k;
    dp[0][0] = 1;
    for(i=1;i<=m + 1;i++){
        for(j=0;j<=m + 1;j++){
            for(int pre=0;pre<j;pre++){
                mint c = pre<k ? pw(j,min(j,k) - pre) : 1;
                dp[i][j] += dp[i - 1][pre]*c;
            }
        }
    }
    f[0] = 1;
    for(i=1;i<=m + 1;i++) f[i] = f[i - 1]*i;
    for(i=1;i<=n;i++) S[i][1] = S[i][i] = 1;
    for(i=2;i<=n;i++){
        for(j=2;j<i;j++) S[i][j] = S[i - 1][j - 1] + j*S[i - 1][j];
    }
    mint ans = 0;
    for(i=1;i<=m;i++){
        // cout << "#empty = " << i << " ";
        // cout << f[m + 1 - i].val() << " " << S[n][m + 1 - i].val() << " " << dp[i][m + 1].val() << endl;
        ans += f[m + 1 - i]*S[n][m + 1 - i]*dp[i][m + 1];
    }
    for(i=m + 2;i<=k;i++) ans *= i;
    cout << ans.val() << "\n";
}
0