結果
問題 | No.2847 Birthday Attack |
ユーザー |
![]() |
提出日時 | 2024-08-23 23:27:17 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 477 ms / 3,000 ms |
コード長 | 771 bytes |
コンパイル時間 | 443 ms |
コンパイル使用メモリ | 82,360 KB |
実行使用メモリ | 76,876 KB |
最終ジャッジ日時 | 2024-08-23 23:27:24 |
合計ジャッジ時間 | 6,677 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 14 |
ソースコード
import math x, y, mod = map(int,input().split()) ans = 0 for p in range(1, 10 ** 7 + 1): for q in range(0, p): if p ** 2 + q ** 2 > max(x, y): break if p % 2 == q % 2: continue if math.gcd(p, q) != 1: continue a = p ** 2 - q ** 2 b = 2 * p * q c = p ** 2 + q ** 2 k = 1 while True: if x - a * k * 2 <= 0: break if y - b * k * 2 <= 0: break ans += (x - 2 * a * k) * (y - 2 * b * k) % mod * ((a > 0) + 1) % mod * ((b > 0) + 1) % mod ans %= mod k += 1 a = 2 * p * q b = p ** 2 - q ** 2 c = p ** 2 + q ** 2 k = 1 while True: if x - a * k * 2 <= 0: break if y - b * k * 2 <= 0: break ans += (x - 2 * a * k) * (y - 2 * b * k) % mod * ((a > 0) + 1) % mod * ((b > 0) + 1) % mod ans %= mod k += 1 print(ans % mod)