結果
問題 | No.2066 Simple Math ! |
ユーザー |
|
提出日時 | 2025-02-06 01:21:29 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 954 ms / 2,000 ms |
コード長 | 1,087 bytes |
コンパイル時間 | 863 ms |
コンパイル使用メモリ | 81,968 KB |
実行使用メモリ | 78,324 KB |
最終ジャッジ日時 | 2025-02-06 01:21:56 |
合計ジャッジ時間 | 25,872 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 31 |
ソースコード
from math import gcd## sum_{i=0}^{n-1}{(i*a + b)//m}def floor_sum(n, m, a, b):ans = 0while True:if a >= m or a < 0:ans += n * (n - 1) * (a // m) // 2a %= mif b >= m or b < 0:ans += n * (b // m)b %= my_max = a * n + bif y_max < m: breakn, b, m, a = y_max // m, y_max % m, a, mreturn ansdef count_ok(p, q, x):# sum_{i=0}^{q-1}((x-i*p)//q + 1) - 1# ただし、 x-i*p >= 0 <=> i <= x // pn = min(q, x//p + 1)return floor_sum(n, q, -p, x) + n - 1def solve():P, Q, K = map(int, input().split())g = gcd(P, Q)p = P // gq = Q // gif 1 in (p, q) and K == 1:return gng = 1ok = P*Kwhile ng+1 < ok:mid = (ng + ok) // 2cnt = count_ok(p, q, mid)# print(f'{p = }, {q = }, {mid = }')# print(f'{cnt = }')if cnt >= K:ok = midelse:ng = midreturn ok * gT = int(input())for _ in range(T):res = solve()print(res)