結果

問題 No.1474 かさまＪ
ユーザー 00 Sakuda00 Sakuda
提出日時 2024-04-04 17:31:05
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 288 ms / 2,500 ms
コード長 1,894 bytes
コンパイル時間 2,208 ms
コンパイル使用メモリ 213,764 KB
実行使用メモリ 16,256 KB
最終ジャッジ日時 2024-04-04 17:31:11
合計ジャッジ時間 5,375 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,676 KB
testcase_01 AC 2 ms
6,676 KB
testcase_02 AC 3 ms
6,676 KB
testcase_03 AC 2 ms
6,676 KB
testcase_04 AC 1 ms
6,676 KB
testcase_05 AC 129 ms
11,520 KB
testcase_06 AC 22 ms
6,676 KB
testcase_07 AC 4 ms
6,676 KB
testcase_08 AC 3 ms
6,676 KB
testcase_09 AC 192 ms
11,904 KB
testcase_10 AC 11 ms
6,676 KB
testcase_11 AC 6 ms
6,676 KB
testcase_12 AC 4 ms
6,676 KB
testcase_13 AC 182 ms
12,928 KB
testcase_14 AC 58 ms
6,676 KB
testcase_15 AC 26 ms
6,676 KB
testcase_16 AC 12 ms
6,676 KB
testcase_17 AC 68 ms
8,852 KB
testcase_18 AC 237 ms
14,976 KB
testcase_19 AC 275 ms
15,928 KB
testcase_20 AC 288 ms
16,256 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/modint>
using namespace std;
using namespace atcoder;
using mint = modint;
using ll = long long;
using VL = vector<ll>;
struct COMB_MOD {
  ll mod;
  int MAX;
  VL fac, finv, inv;
  COMB_MOD(int max, ll m) {
    fac.assign(max, 0);
    finv.assign(max, 0);
    inv.assign(max, 0);
    mod = m;
    MAX = max;
  }
  void solve() {
    fac[0] = fac[1] = 1;
    finv[0] = finv[1] = 1;
    inv[1] = 1;
    for (int i = 2; i < MAX; i++){
        fac[i] = fac[i - 1] * i % mod;

        inv[i] = mod - inv[mod%i] * (mod / i) % mod;
        finv[i] = finv[i - 1] * inv[i] % mod;
    }
    
  }
  ll comb(int n, int k) {
    if (n < k) return 0;
    if (n < 0 || k < 0) return 0;
    return fac[n] * (finv[k] * finv[n - k] % mod) % mod;
  }
};

int main() {
	mint::set_mod((ll)1e9 + 7);
	int N, M, MP, MQ, L;
	cin >> N >> MP >> MQ >> L;
	vector<vector<mint>>  dp(N + 1, vector<mint>(MQ + 1, 0));
	dp[0][0] = 1;
	vector<vector<mint>> sdp(N + 1, vector<mint>(MQ + 1, 0));
	for (int i = 0;i <= MQ;i++) sdp[0][i] = 1;
	for (int i = 1;i <= N;i++) {
		int s;cin >> s;
		vector<vector<mint>> ndp(N + 1, vector<mint>(MQ + 1, 0));
		vector<vector<mint>> nsdp(N + 1, vector<mint> (MQ + 1, 0));
		for (int j = 0;j <= N;j++) for (int k = 0;k <= MQ;k++) {
			ndp[j][k] += dp[j][k];
			if (k == 0) {
				nsdp[j][k] += ndp[j][k];
			} else {
				nsdp[j][k] = nsdp[j][k - 1] + ndp[j][k];
			}
			if (j == 0 or k == 0) continue;
			int l = k - min(s, k);int r = k - 1;
			ndp[j][k] += sdp[j - 1][r] - (l == 0 ? 0 : sdp[j - 1][l - 1]);
			nsdp[j][k] = nsdp[j][k - 1] + ndp[j][k];
		}
		swap(dp, ndp);
		swap(sdp, nsdp);
	}
	COMB_MOD comb(MP + MQ + N + 100, (ll)(1e9 + 7));
	comb.solve();
	mint ans = 0;
	for (int m = 0;m <= MQ;m++) for (int n = 0;n <= N;n++) {
		int R = MP - (L * n) + m;
		ans += mint(comb.comb(R + N - 1, N - 1)) * dp[n][m];
	}
	cout << ans.val() << endl;
}
0