結果
問題 | No.2406 Difference of Coordinate Squared |
ユーザー |
![]() |
提出日時 | 2023-08-04 22:17:32 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 321 ms / 2,000 ms |
コード長 | 1,459 bytes |
コンパイル時間 | 1,427 ms |
コンパイル使用メモリ | 169,744 KB |
実行使用メモリ | 26,624 KB |
最終ジャッジ日時 | 2024-11-26 18:02:37 |
合計ジャッジ時間 | 5,619 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 55 |
ソースコード
#include <bits/stdc++.h>using namespace std;const long long MOD = 998244353;const long long HALF = 499122177;int main(){int N;long long M;cin >> N >> M;vector<long long> inv(N + 1);inv[1] = 1;for (int i = 2; i <= N; i++){inv[i] = MOD - inv[MOD % i] * (MOD / i) % MOD;}vector<long long> fact(N + 1), finv(N + 1);fact[0] = 1;finv[0] = 1;for (int i = 1; i <= N; i++){fact[i] = fact[i - 1] * i % MOD;finv[i] = finv[i - 1] * inv[i] % MOD;}long long P = 1;for (int i = 0; i < N * 2; i++){P *= HALF;P %= MOD;}long long ans = 0;for (long long Y = -N; Y <= N; Y++){long long XX = Y * Y + M;long long tv = 0, fv = 2000000;while (fv - tv > 1){long long mid = (tv + fv) / 2;if (mid * mid <= XX){tv = mid;} else {fv = mid;}}if (XX == tv * tv){long long X = tv;long long S = X + Y;long long D = X - Y;if ((N - S) % 2 == 0 && (N - D) % 2 == 0){if (-N <= S && S <= N && -N <= D && D <= N){long long add = P;add *= fact[N] * finv[(N + S) / 2] % MOD * finv[(N - S) / 2] % MOD;add %= MOD;add *= fact[N] * finv[(N + D) / 2] % MOD * finv[(N - D) / 2] % MOD;add %= MOD;if (X > 0){add *= 2;add %= MOD;}ans += add;}}}}ans %= MOD;cout << ans << endl;}