結果

問題 No.800 四平方定理
コンテスト
ユーザー yurujirou
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#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;
}
0