結果
問題 | No.2362 Inversion Number of Mod of Linear |
ユーザー | 遭難者 |
提出日時 | 2023-06-18 12:27:10 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 479 ms / 2,000 ms |
コード長 | 1,128 bytes |
コンパイル時間 | 285 ms |
コンパイル使用メモリ | 87,052 KB |
実行使用メモリ | 80,128 KB |
最終ジャッジ日時 | 2023-09-08 08:07:46 |
合計ジャッジ時間 | 3,028 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge13 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 74 ms
70,960 KB |
testcase_01 | AC | 74 ms
71,080 KB |
testcase_02 | AC | 73 ms
71,076 KB |
testcase_03 | AC | 324 ms
79,880 KB |
testcase_04 | AC | 143 ms
77,776 KB |
testcase_05 | AC | 157 ms
78,152 KB |
testcase_06 | AC | 479 ms
80,128 KB |
testcase_07 | AC | 271 ms
79,556 KB |
testcase_08 | AC | 197 ms
78,704 KB |
testcase_09 | AC | 197 ms
78,092 KB |
ソースコード
import sys input = sys.stdin.buffer.readline from math import gcd def q(n, m, a, b, c=0): if m == 0: return 0, 0, 0 a1, a2 = a // m, a % m b1, b2 = b // m, b % m y = (a2 * n + b2) // m ff, gg, hh = q(y, a2, m, m + a2 - b2 - 1, c + 1) nn = n * (n - 1) // 2 f = n * y - ff g = n * y * y - ff - 2 * hh h = nn * y + (ff - gg) // 2 g += n * (2 * n - 1) * (n - 1) * a1 * a1 // 6 g += 2 * nn * a1 * b1 g += b1 * b1 * n g += 2 * a1 * h g += 2 * b1 * f f += a1 * nn + b1 * n h += nn * (a1 * (2 * n - 1) + 3 * b1) // 3 return f, g, h def main() -> None: n, m, a, b = map(int, input().split()) gccd = gcd(m, a) m, a, b = m // gccd, a // gccd, b // gccd ans = 0 f, g, h = q(n, m, a, b) ans -= (n - 1) * f ans += 2 * h t = a * (n - 1) // m y = m - a * (n - 1) + t * m f, g, h = q(n, m, a, y) ans += f ans += h ans -= n * (n + 1) // 2 * t k1, k2 = n // m, n % m ans -= (k1 + 1) * (k1 + 2) // 2 * k2 ans -= k1 * (k1 + 1) // 2 * (m - k2) print(ans) return for i in range(int(input())): main()