結果

問題 No.800 四平方定理
ユーザー ohreitetsu
提出日時 2019-04-25 22:06:29
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 885 bytes
コンパイル時間 745 ms
コンパイル使用メモリ 75,648 KB
実行使用メモリ 167,680 KB
最終ジャッジ日時 2024-11-21 14:39:42
合計ジャッジ時間 44,804 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 TLE * 1
other AC * 21 TLE * 9
権限があれば一括ダウンロードができます

ソースコード

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++;
			}
		}
	}
	mit=xyMap.begin();
	int MinV=(*mit).first;
	for (int w=1;w<=N;w++){
		for (int z=1;z<=N;z++){
			int wz=w*w-z*z+D;
			if (wz<MinV){
				continue;
			}
			mit=wzMap.find(wz);
			if (mit==wzMap.end()){
				wzMap[wz]=1;
			}else{
				(*mit).second++;
			}
		}
	}
	int ans=0;
	map<int,int>::iterator mit1;
	for (mit=xyMap.begin();mit!=xyMap.end();mit++){
		mit1=wzMap.find((*mit).first);
		if (mit1!=wzMap.end()){
			ans+=(*mit).second * (*mit1).second;
		}
	}
	xyMap.clear();
	wzMap.clear();
	cout<<ans<<endl;		
	return 0;
}
0