結果

問題 No.1474 かさまＪ
ユーザー hitonanode
提出日時 2021-04-09 22:17:49
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 66 ms / 2,500 ms
コード長 1,276 bytes
コンパイル時間 3,393 ms
コンパイル使用メモリ 202,148 KB
最終ジャッジ日時 2025-01-20 14:34:37
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define FOR(i, begin, end) for(int i=(begin),i##_end_=(end);i<i##_end_;i++)
#define IFOR(i, begin, end) for(int i=(end)-1,i##_begin_=(begin);i>=i##_begin_;i--)
#define REP(i, n) FOR(i,0,n)
#define IREP(i, n) IFOR(i,0,n)

#include <atcoder/modint>
using mint = atcoder::modint1000000007;

int main() {
    int N, mp, mq, L;
    cin >> N >> mp >> mq >> L;
    vector<int> S(N);
    for (auto &x : S) cin >> x;
    vector cnt(N + 1, vector<mint>(mq + 1));
    cnt[0][0] = 1;
    REP(k, N) {
        auto &s = S[k];
        IREP(i, k + 1) {
            mint cum = 0;
            REP(j, mq + 1) {
                cnt[i + 1][j] += cum, cum += cnt[i][j] - (j - s >= 0 ? cnt[i][j - s] : 0);
            }
        }
    }

    mint ret = 0;
    vector<mint> fac(N + mp + 1, 1), facinv(N + mp + 1);
    FOR(i, 1, fac.size()) fac[i] = fac[i - 1] * i;
    facinv.back() = 1 / fac.back();
    IREP(i, facinv.size() - 1) facinv[i] = facinv[i + 1] * (i + 1);

    REP(sz, cnt.size()) REP(i, cnt[sz].size()) {
        if (cnt[sz][i].val()) {
            int rem = mp - (sz * L - i);
            if (rem < 0) continue;
            ret += cnt[sz][i] * fac[rem + N - 1] * facinv[rem];
        }
    }
    cout << (ret / fac[N - 1]).val() << '\n';
}
0