結果
問題 | No.2847 Birthday Attack |
ユーザー | 👑 rin204 |
提出日時 | 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 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 38 ms
52,132 KB |
testcase_01 | AC | 38 ms
52,588 KB |
testcase_02 | AC | 35 ms
52,588 KB |
testcase_03 | AC | 302 ms
76,484 KB |
testcase_04 | AC | 37 ms
52,952 KB |
testcase_05 | AC | 982 ms
76,660 KB |
testcase_06 | AC | 533 ms
76,724 KB |
testcase_07 | AC | 912 ms
76,540 KB |
testcase_08 | AC | 306 ms
76,584 KB |
testcase_09 | AC | 609 ms
76,772 KB |
testcase_10 | AC | 447 ms
76,804 KB |
testcase_11 | AC | 538 ms
76,636 KB |
testcase_12 | AC | 826 ms
76,664 KB |
testcase_13 | AC | 991 ms
76,584 KB |
testcase_14 | AC | 695 ms
76,656 KB |
testcase_15 | AC | 769 ms
76,456 KB |
testcase_16 | AC | 611 ms
76,632 KB |
testcase_17 | AC | 808 ms
76,808 KB |
ソースコード
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)