結果
問題 |
No.2847 Birthday Attack
|
ユーザー |
👑 |
提出日時 | 2024-07-09 23:12:29 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 548 ms / 3,000 ms |
コード長 | 870 bytes |
コンパイル時間 | 261 ms |
コンパイル使用メモリ | 82,148 KB |
実行使用メモリ | 76,672 KB |
最終ジャッジ日時 | 2024-07-09 23:12:42 |
合計ジャッジ時間 | 6,752 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 14 |
ソースコード
from math import gcd x,y,md = map(int, input().split()) assert 1 <= x <= 4_000_000 and 1 <= y <= 4_000_000 and 100_000_000 <= md <= 1_000_000_000 ans = 0 mx = max(x, y) // 2 + 1 for m in range(0, mx+1): for n in range(m+1, mx+1): if (m % 2 == n % 2) or gcd(m, n) != 1: continue ai = n*n - m*m bi = 2*m*n if (ai > mx) or (bi > mx): break mul = 1 while (mul * max(ai, bi) <= mx): a = ai*mul b = bi*mul if a*2 < x and b*2 < y: if (b == 0): ans = (ans + (x - 2*a) * (y - 2*b) * 2) % md else: ans = (ans + (x - 2*a) * (y - 2*b) * 4) % md if b*2 < x and a*2 < y: if (b == 0): ans = (ans + (x - 2*b) * (y - 2*a) * 2) % md else: ans = (ans + (x - 2*b) * (y - 2*a) * 4) % md mul += 1 print(ans % md)