結果

問題 No.1474 かさまＪ
ユーザー hitonanode
提出日時 2021-04-09 22:11:54
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 175 ms / 2,500 ms
コード長 1,258 bytes
コンパイル時間 2,079 ms
コンパイル使用メモリ 200,808 KB
最終ジャッジ日時 2025-01-20 14:30:24
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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, 0));
    cnt[0][0] = 1;
    for (auto s : S) {
        IREP(i, cnt.size() - 1) {
            vector<mint> nxt(cnt[i + 1].size() + s + 1);
            REP(j, cnt[i].size()) {
                nxt[j + 1] += cnt[i][j];
                nxt[j + s + 1] -= cnt[i][j];
            }
            REP(j, cnt[i].size()) nxt[j + 1] += nxt[j], cnt[i + 1][j] += nxt[j];
        }
    }

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

    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] / (fac[rem] * fac[N - 1]);
        }
    }
    cout << ret.val() << '\n';
}
0