結果

問題 No.1474 かさまＪ
ユーザー gorugo30
提出日時 2021-03-02 17:25:32
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 750 ms / 2,500 ms
コード長 979 bytes
コンパイル時間 196 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 83,712 KB
最終ジャッジ日時 2024-10-03 01:52:54
合計ジャッジ時間 6,568 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

N, MP, Mq, L = map(int, input().split())
S = list(map(int, input().split()))
MOD = 10 ** 9 + 7

dp = [[0] * (Mq + 1) for k in range(N + 1)]
dp[0][0] = 1

for i in range(N):
    s = S[i]
    for n in range(N - 1, -1, -1):
        for m in range(Mq + 1):
            if m + 1 <= Mq:
                dp[n + 1][m + 1] = (dp[n + 1][m + 1] + dp[n][m]) % MOD
            if m + s + 1 <= Mq:
                dp[n + 1][m + s + 1] = (dp[n + 1][m + s + 1] - dp[n][m]) % MOD
for n in range(N + 1):
    for i in range(n):
        for m in range(1, Mq + 1):
            dp[n][m] = (dp[n][m] + dp[n][m - 1]) % MOD

fac = [1] * (50000)
inv = [1] * (50000)
for i in range(1, 50000):
    fac[i] = (fac[i - 1] * i) % MOD
    inv[i] = pow(fac[i], MOD - 2, MOD)
def nCr(n, r):
    if n < r:
        return 0
    return (fac[n] * inv[n - r] * inv[r]) % MOD

ans = 0
for m in range(Mq + 1):
    for n in range(N + 1):
        ans = (ans + dp[n][m] * nCr(MP + m - L * n + N - 1, N - 1)) % MOD
print(ans)
0