結果
問題 | No.2474 Empty Quartz |
ユーザー | ntuda |
提出日時 | 2023-10-01 15:35:07 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 397 ms / 2,000 ms |
コード長 | 875 bytes |
コンパイル時間 | 331 ms |
コンパイル使用メモリ | 87,356 KB |
実行使用メモリ | 81,892 KB |
最終ジャッジ日時 | 2023-10-01 15:35:11 |
合計ジャッジ時間 | 3,220 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 78 ms
77,872 KB |
testcase_01 | AC | 76 ms
78,112 KB |
testcase_02 | AC | 345 ms
80,924 KB |
testcase_03 | AC | 306 ms
80,336 KB |
testcase_04 | AC | 376 ms
81,892 KB |
testcase_05 | AC | 391 ms
80,560 KB |
testcase_06 | AC | 397 ms
80,236 KB |
ソースコード
MOD = 998244353 N = 10 ** 5 + 10 fact = [1] * (N + 1) rfact = [1] * (N + 1) r = 1 for i in range(1, N + 1): fact[i] = r = r * i % MOD rfact[N] = r = pow(fact[N], MOD - 2, MOD) for i in range(N, 0, -1): rfact[i - 1] = r = r * i % MOD def comb(n, k): return fact[n] * rfact[k] * rfact[n - k] % MOD T = int(input()) for _ in range(T): N, K = map(int, input().split()) if N == 1: print(N) continue if K == 0: print(1) continue if (N + 1) ** 2 - 4 * K < 0: print(0) continue c = (N + 1) ** 2 - 4 * K d = int(c ** 0.5) for i in range(d, d + 2): if i * i == c: break else: print(0) continue ans = 0 a = (N + 1 + i) // 2 ans += comb(N, a) if i != 0: a = (N + 1 - i) // 2 ans += comb(N, a) ans %= MOD print(ans)