結果

問題 No.800 四平方定理
ユーザー BwambocosBwambocos
提出日時 2019-03-17 23:15:08
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 621 bytes
コンパイル時間 1,545 ms
コンパイル使用メモリ 168,300 KB
実行使用メモリ 10,148 KB
最終ジャッジ日時 2024-07-08 02:19:23
合計ジャッジ時間 5,385 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample -- * 3
other AC * 10 TLE * 1 -- * 19
権限があれば一括ダウンロードができます

ソースコード

diff #

#include "bits/stdc++.h"
#define in std::cin
#define out std::cout
#define rep(i,N) for(LL i=0;i<N;++i)
typedef long long int LL;

int main()
{
	LL N, D;
	in >> N >> D;

	LL ans = 0, max = N * N + D;
	LL maxx = std::min<LL>(N, sqrt(max));
	for (LL x = 1; x <= maxx; ++x)
	{
		if (x * x > max) continue;
		for (LL y = 1; y <= maxx; ++y)
		{
			if (x * x + y * y > max) continue;
			for (LL z = 1; z <= maxx; ++z)
			{
				if (x * x + y * y + z * z > max) continue;
				LL w2 = x * x + y * y + z * z - D;
				if (w2 > 0)
				{
					LL w = sqrt(w2);
					if (w*w == w2) ++ans;
				}
			}
		}
	}

	out << ans << std::endl;
}
0