結果
問題 |
No.1201 お菓子配り-4
|
ユーザー |
|
提出日時 | 2023-03-11 15:14:27 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,002 bytes |
コンパイル時間 | 446 ms |
コンパイル使用メモリ | 82,824 KB |
実行使用メモリ | 82,484 KB |
最終ジャッジ日時 | 2024-09-18 06:25:56 |
合計ジャッジ時間 | 50,372 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 30 TLE * 6 |
ソースコード
from math import ceil MOD = int(1e9 + 7) def floor_sum_of_linear(L: int, R: int, a: int, b: int, div: int) -> int: """ ``` sum((x * a + b) // div for x in range(L, R)) ``` """ if L >= R: return 0 res = 0 b += L * a n = R - L if b < 0: k = ceil(-b / div) b += k * div res -= n * k while n: q, a = a // div, a % div res += n * (n - 1) // 2 * q # res %= MOD if b >= div: q, b = b // div, b % div res += n * q # res %= MOD n, b = (a * n + b) // div, (a * n + b) % div a, div = div, a return res if __name__ == "__main__": n = int(input()) m = int(input()) nums1 = list(map(int, input().split())) nums2 = list(map(int, input().split())) res = 0 for a in nums1: for b in nums2: res += floor_sum_of_linear(1, b + 1, a, 0, b) res %= MOD res *= 2 res %= MOD print(res)