結果
問題 | No.800 四平方定理 |
ユーザー |
|
提出日時 | 2023-04-04 15:33:17 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 105 ms / 2,000 ms |
コード長 | 905 bytes |
コンパイル時間 | 4,014 ms |
コンパイル使用メモリ | 251,604 KB |
最終ジャッジ日時 | 2025-02-11 22:53:33 |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
#define _CRT_SECURE_NO_WARNINGS #define _USE_MATH_DEFINES #include <bits/stdc++.h> using namespace std; #include <atcoder/all> using namespace atcoder; #define REP(i, n) for(int i = 0; i < (int)n; i++) #define LREP(i, n) for(LL i = 0; i < (LL)n; i++) #define RREP(i, n) for(int i = (int)n-1; i >= 0; i--) #define V(T) vector<T> #define P pair<int, int> #define LP pair<LL, LL> #define T3 tuple<LL, LL, LL> #define T4 tuple<LL, LL, LL, LL> #define INF 1000000007 #define SIZE 500100 #define MOD 1000000007 typedef long long LL; #define SIZE2 2010 LL N, D; LL C[2 * SIZE2 * SIZE2]; int main() { cin >> N >> D; for (LL x = 1; x <= N; x++) { for (LL y = 1; y <= N; y++) { C[x * x + y * y] += 1; } } LL ans = 0; for (LL z = 1; z <= N; z++) { for (LL w = 1; w <= N; w++) { LL Y = D + w * w - z * z; if (0 <= Y && Y <= 2 * N * N) { ans += C[Y]; } } } cout << ans << endl; }