結果

問題 No.2951 Similar to Mex
ユーザー pockynypockyny
提出日時 2024-10-26 18:20:21
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,083 bytes
コンパイル時間 946 ms
コンパイル使用メモリ 73,008 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-10-26 18:20:29
合計ジャッジ時間 7,852 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28 WA * 14
権限があれば一括ダウンロードができます

ソースコード

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];
    }
    cout << ans.val() << "\n";
}
0