結果

問題 No.1238 選抜クラス
ユーザー nawawan
提出日時 2020-09-26 12:21:07
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 18 ms / 2,000 ms
コード長 920 bytes
コンパイル時間 689 ms
コンパイル使用メモリ 73,576 KB
実行使用メモリ 19,584 KB
最終ジャッジ日時 2024-06-28 22:46:14
合計ジャッジ時間 2,169 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
using namespace std;
int main(){
    int N, K;
    const int MOD = 1e9 + 7;
    cin >> N >> K;
    vector<int> A(N);
    for(int i = 0; i < N; i++) {
        cin >> A[i];
        A[i] -= K;
    }
    vector<vector<long long>> dp(N + 1, vector<long long> (20010, -1));
    dp[0][10000] = 1;
    for(int i = 0; i < N; i++){
        for(int j = 0; j < 20010; j++){
            if(dp[i][j] == -1) continue;
            if(dp[i + 1][j] == -1) {
                dp[i + 1][j] = dp[i][j];
            }
            else dp[i + 1][j] = (dp[i + 1][j] + dp[i][j]) % MOD;
            if(dp[i + 1][j + A[i]] == -1) dp[i + 1][j + A[i]] = dp[i][j];
            else dp[i + 1][j + A[i]] = (dp[i + 1][j + A[i]] + dp[i][j]) % MOD;
        }
    }
    long long ans = -1;
    for(int i = 10000; i < 20010; i++){
        if(dp[N][i] != -1) ans = (ans + dp[N][i]) % MOD;
    }
    cout << ans << endl;
}
0