結果

問題 No.1899 L1 Cafe
ユーザー 👑 rin204rin204
提出日時 2022-04-08 22:21:15
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 417 ms / 2,000 ms
コード長 860 bytes
コンパイル時間 348 ms
コンパイル使用メモリ 82,452 KB
実行使用メモリ 78,132 KB
最終ジャッジ日時 2024-11-28 13:02:59
合計ジャッジ時間 5,987 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

# return (\sum_{i=0}^{n-1} ((a*i+b)//m))
def floor_sum(n, m, a, b):
    ret = 0
    while True:
        if a >= m:
            ret += ((n - 1) * n) // 2 * (a // m)
            a %= m
        if b >= m:
            ret += n * (b // m)
            b %= m
        y_max = (a * n + b) // m
        if y_max == 0:
            return ret
        x_max = y_max * m - b
        ret += (n - (x_max + a - 1) // a) * y_max
        n, m, a, b = y_max, a, m, -x_max % a

def solve():
    n, a, b = map(int, input().split())
    ans = 0
    max_ = n - a
    min_ = n % a
    if max_ > 0:
        ans += (max_ + min_) * (max_ - min_ + a) // a // 2
        
    max_ = n - b
    min_ = n % b
    if max_ > 0:
        ans += (max_ + min_) * (max_ - min_ + b) // b // 2
    
    ans -= floor_sum(n // a, b, a, n % a)
    print(ans)
    
for _ in range(int(input())):
    solve()
0