結果

問題 No.1693 Invasion
ユーザー eSeFeSeF
提出日時 2021-08-08 01:43:11
言語 C++17(clang)
(17.0.6 + boost 1.87.0)
結果
AC  
実行時間 14 ms / 2,000 ms
コード長 1,125 bytes
コンパイル時間 8,648 ms
コンパイル使用メモリ 180,992 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-07-19 09:43:45
合計ジャッジ時間 9,525 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
const int INF = 1073741823;

using mint = modint998244353;
vector<mint> fac = vector<mint>();
vector<mint> faci = vector<mint>();
void init_mod(int n){
    assert(0 < n);
    fac.push_back(1);
    for(int i = 1; i <= n; i++) fac.push_back(fac.back() * i);
    faci.push_back(fac.back().inv());
    for(int i = n; i > 0; i--) faci.push_back(faci.back() * i);
    reverse(faci.begin(), faci.end());
}
mint binom(int n,int r){
    if(n < 0 || r < 0 || n < r) return 0;
    return fac[n] * faci[r] * faci[n - r];
}

int main(){
    int N, M;
    cin >> N >> M;
    vector<int>A(N);
    for(int i = 0; i < N; i++) cin >> A[i];
    vector<int>dp(M + 1, INF);
    dp[0] = 0;
    for(int i = 0; i < M; i++){
        for(int j = 0; j < N; j++){
            if(i + A[j] <= M) dp[i + A[j]] = min(dp[i + A[j]], dp[i] + 1);
        }
    }
    init_mod(M);
    mint ans = 0;
    for(int K = 0; K <= M; K++){
        if(dp[K] == INF) continue;
        ans += binom(M - dp[K], K - dp[K]);
    }
    cout << ans.val() << endl;
    
    return 0;
}
0