結果
問題 | No.2847 Birthday Attack |
ユーザー |
👑 |
提出日時 | 2024-08-23 23:38:05 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 991 ms / 3,000 ms |
コード長 | 1,219 bytes |
コンパイル時間 | 273 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 76,808 KB |
最終ジャッジ日時 | 2024-08-23 23:38:17 |
合計ジャッジ時間 | 10,795 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 14 |
ソースコード
from math import gcd x, y, MOD = map(int, input().split()) ans0 = 0 ox = x // 2 ex = x - ox oy = y // 2 ey = y - oy ans0 += ox * (ox - 1) // 2 * y ans0 += ex * (ex - 1) // 2 * y ans0 += oy * (oy - 1) // 2 * x ans0 += ey * (ey - 1) // 2 * x ans0 %= MOD ans = 0 for n in range(1, max(x, y) + 1): for m in range(n + 1, max(x, y) + 1): if (m - n) % 2 == 0 or gcd(m, n) != 1: continue a = m * m - n * n b = 2 * m * n c = m * m + n * n a *= 2 b *= 2 c *= 2 if b > max(x, y) or c > 10**9: break k = min(x // a, y // b) ans += x * y * k ans %= MOD ans -= b * x * (k * (k + 1) // 2) % MOD ans %= MOD ans -= a * y * (k * (k + 1) // 2) % MOD ans %= MOD ans += a * b * (k * (k + 1) * (2 * k + 1) // 6) % MOD ans %= MOD k = min(x // b, y // a) ans += x * y * k ans %= MOD ans -= a * x * (k * (k + 1) // 2) % MOD ans %= MOD ans -= b * y * (k * (k + 1) // 2) % MOD ans %= MOD ans += a * b * (k * (k + 1) * (2 * k + 1) // 6) % MOD ans %= MOD ans *= 2 ans += ans0 ans *= 2 print(ans % MOD)