結果

問題 No.1043 直列大学
ユーザー pekempey
提出日時 2020-05-01 21:58:27
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 986 ms / 2,000 ms
コード長 662 bytes
コンパイル時間 201 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 44,448 KB
最終ジャッジ日時 2024-12-22 19:28:28
合計ジャッジ時間 30,491 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff #

def input_int():
    return int(input())

def input_ints():
    return list(map(int, input().split()))

import numpy as np

MOD = 10**9 + 7

N, M = input_ints()
V = input_ints()
R = input_ints()
A, B = input_ints()

dp = np.zeros(10**5 + 10, dtype=np.int64)
dp[0] = 1
for i in range(N):
    dp[V[i]:] += dp[:-V[i]]
    dp %= MOD

ep = np.zeros(10**5 + 10, dtype=np.int64)
ep[0] = 1
for i in range(M):
    ep[R[i]:] += ep[:-R[i]]
    ep %= MOD
ep[0] = 0
ep = np.cumsum(ep)
ep %= MOD

ans = 0
for i in range(len(dp)):
    # i/B <= j <= i/A
    ans += dp[i] * ep[i // A]
    if i > 0:
        ans -= dp[i] * ep[(i + B - 1) // B - 1]
    ans %= MOD

print(ans % MOD)
0