結果

問題 No.800 四平方定理
コンテスト
ユーザー ohreitetsu
提出日時 2019-04-25 22:52:00
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,075 bytes
コンパイル時間 811 ms
コンパイル使用メモリ 76,136 KB
実行使用メモリ 140,160 KB
最終ジャッジ日時 2024-11-21 15:59:15
合計ジャッジ時間 51,287 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 TLE * 1
other AC * 16 TLE * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <map>
using namespace std;
int main(int argc, char* argv[])
{
	int N,D;
	cin>>N>>D;
	map<int,int> xyMap;
	map<int,int> wzMap;
	map<int,int>::iterator mit;
	for (int x=1;x<=N;x++){
		for (int y=1;y<=N;y++){
			int xy=x*x+y*y;
			mit=xyMap.find(xy);
			if (mit==xyMap.end()){
				xyMap[xy]=1;
			}else{
				(*mit).second++;
			}
			int wz=x*x-y*y+D;
			if (wz<2){
				continue;
			}
			mit=wzMap.find(wz);
			if (mit==wzMap.end()){
				wzMap[wz]=1;
			}else{
				(*mit).second++;
			}
		}
	}
	int ans=0;
	map<int,int>::iterator mit1;
	while (true){
		mit=xyMap.begin();
		if (mit==xyMap.end()){
			break;
		}
		while (true){
			mit1=wzMap.begin();
			if (mit1==wzMap.end()){
				break;
			}
			if ((*mit).first>(*mit1).first){
				wzMap.erase(mit1);
			}else if ( (*mit).first==(*mit1).first){
				ans+=(*mit).second * (*mit1).second;
				wzMap.erase(mit1);
				break;
			}else{
				break;
			}
		}
		mit1=wzMap.begin();
		if (mit1==wzMap.end()){
			break;
		}
		xyMap.erase(mit);
	}
	xyMap.clear();
	wzMap.clear();
	cout<<ans<<endl;		
	return 0;
}
0