結果

問題 No.800 四平方定理
ユーザー e869120
提出日時 2019-05-15 21:15:02
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 827 ms / 2,000 ms
コード長 616 bytes
コンパイル時間 983 ms
コンパイル使用メモリ 76,344 KB
実行使用メモリ 35,160 KB
最終ジャッジ日時 2024-09-14 13:10:53
合計ジャッジ時間 13,488 ms
ジャッジサーバーID
(参考情報)
judge6 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

long long N, D, sum; vector<int>E1, E2;

int main() {
	cin >> N >> D;
	for (int i = 1; i <= N; i++) {
		for (int j = 1; j <= N; j++) {
			E1.push_back(i*i + j * j);
			E2.push_back(i*i - j * j);
		}
	}
	sort(E1.begin(), E1.end());
	sort(E2.begin(), E2.end());

	for (int i = 0; i < E1.size(); i++) {
		int target = D - E1[i];
		int pos1 = lower_bound(E2.begin(), E2.end(), target) - E2.begin();
		int pos2 = lower_bound(E2.begin(), E2.end(), target + 1) - E2.begin();
		sum += 1LL * (pos2 - pos1);
	}
	cout << sum << endl;
	return 0;
}
0