結果

問題 No.800 四平方定理
ユーザー tempura_pp
提出日時 2019-03-14 14:51:36
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
(最新)
AC  
(最初)
実行時間 -
コード長 448 bytes
コンパイル時間 2,030 ms
コンパイル使用メモリ 167,828 KB
実行使用メモリ 81,408 KB
最終ジャッジ日時 2024-07-01 19:37:28
合計ジャッジ時間 6,434 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22 RE * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#define REP(i,m,n) for(int i=(int)(m) ; i < (int) (n) ; ++i )
#define rep(i,n) REP(i,0,n)

int MAX = 5e6;
int main(){
	int n;
	cin>>n;
	int d;
	cin>>d;
	assert(1<=n&&n<=1500);
	assert(0<=d&&d<=1000000);
	vector<long long> cnt1(MAX),cnt2(MAX);
	REP(i,1,n+1)REP(j,1,n+1){
		cnt1[i*i+j*j]++;
		if(i*i-j*j+d>0)cnt2[i*i-j*j+d]++;
	}
	long long ans=0;
	rep(i,MAX)ans+=cnt1[i]*cnt2[i];
	cout<<ans;
	return 0;
}
0