結果

問題 No.1043 直列大学
ユーザー neterukun
提出日時 2020-05-01 21:53:51
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 179 ms / 2,000 ms
コード長 1,070 bytes
コンパイル時間 311 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 69,504 KB
最終ジャッジ日時 2024-12-22 19:26:37
合計ジャッジ時間 3,802 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff #

n, m = map(int, input().split())
v = list(map(int, input().split()))
r = list(map(int, input().split()))
a, b = map(int, input().split())
MOD = 10 ** 9 + 7

sum_v = sum(v)
dv = [0] * (sum_v + 1)
dv[0] = 1
for i in range(n):
    val = v[i]
    for w in range(sum_v + 1)[::-1]:
        if w - val >= 0:
            dv[w] += dv[w - val]
            dv[w] %= MOD
            
sum_r = sum(r)
dr = [0] * (sum_r + 1)
dr[0] = 1
for i in range(m):
    val = r[i]
    for w in range(sum_r + 1)[::-1]:
        if w - val >= 0:
            dr[w] += dr[w - val]
            dr[w] %= MOD

ru_v = [0] * (sum_v + 2)
for i in range(sum_v + 1):
    ru_v[i + 1] = ru_v[i] + dv[i]
    ru_v[i + 1] %= MOD

# A
ptn_a = 0
for i in range(1, sum_r + 1):
    val = i * a
    # ptn_a += sum(dv[1:val]) * dr[i]
    ptn_a += (ru_v[min(val, sum_v + 1)] - ru_v[1]) * dr[i]
    ptn_a %= MOD
# B
ptn_b = 0
for i in range(1, sum_r + 1):
    val = i * b
    # ptn_b += sum(dv[1:val + 1]) * dr[i]
    ptn_b += (ru_v[min(val + 1, sum_v + 1)] - ru_v[1]) * dr[i]
    ptn_b %= MOD

print((ptn_b - ptn_a) % MOD)
0