結果

問題 No.1201 お菓子配り-4
ユーザー rlangevinrlangevin
提出日時 2023-10-26 23:20:17
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 724 bytes
コンパイル時間 493 ms
コンパイル使用メモリ 82,272 KB
実行使用メモリ 796,120 KB
最終ジャッジ日時 2024-09-25 12:32:47
合計ジャッジ時間 10,831 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other TLE * 1 MLE * 2 -- * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

# https://qiita.com/AkariLuminous/items/3e2c80baa6d5e6f3abe9#4-floor_sum
mod = 10**9 + 7
from functools import lru_cache

@lru_cache(maxsize=None)
def floor_sum(n, m, a, b=0):
    ans = 0
    while True:
        if a >= m or a < 0:
            ans += n * (n - 1) * (a // m) // 2
            a %= m
        if b >= m or b < 0:
            ans += n * (b // m)
            b %= m
        y_max = a * n + b
        if y_max < m: break
        n, b, m, a = y_max // m, y_max % m, a, m
    return ans   

N = int(input())
M = int(input())
A = list(map(int, input().split()))
B = list(map(int, input().split()))

ans = 0
for a in A:
    for b in B:
        ans += floor_sum(b + 1, b, a)
    ans %= mod
        
print(ans * 2 % mod)
0