結果

問題 No.1238 選抜クラス
ユーザー srjywrdnprkt
提出日時 2023-01-23 02:57:26
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 645 ms / 2,000 ms
コード長 1,083 bytes
コンパイル時間 1,056 ms
コンパイル使用メモリ 110,692 KB
最終ジャッジ日時 2025-02-10 06:28:07
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <cmath>
#include <map>
#include <set>
#include <iomanip>
#include <queue>
#include <algorithm>
#include <numeric>
#include <deque>
#include <complex>
#include <cassert>

using namespace std;

const long long modc = 1e9+7;

int main(){

    long long N, K, ans=0;
    cin >> N >> K;

    vector<long long> A(N+1);
    for (int i=1; i<=N; i++) cin >> A[i];

    vector<vector<long long>> dp(N+1, vector<long long>(10001));
    dp[0][0] = 1;

    for (int i=1; i<=N; i++){
        vector<vector<long long>> pd(N+1, vector<long long>(10001));
        for (int j=0; j<=N; j++){
            for (int k=0; k<=10000; k++){
                pd[j][k] += dp[j][k];
                if (j>=1 && k>=A[i]) pd[j][k] += dp[j-1][k-A[i]];
                pd[j][k] %= modc;
            }
        }
        swap(dp, pd);
    }

    for (int i=1; i<=N; i++){
        for (int j=0; j<=10000; j++){
            if (j >= K*i){
                ans += dp[i][j];
                ans %= modc;
            }
        }
    }

    cout << ans << endl;

    return 0;
}
0