結果

問題 No.800 四平方定理
ユーザー hanbei_dayo
提出日時 2020-08-11 19:52:16
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,251 ms / 2,000 ms
コード長 1,083 bytes
コンパイル時間 973 ms
コンパイル使用メモリ 103,860 KB
実行使用メモリ 43,468 KB
最終ジャッジ日時 2024-10-09 11:38:56
合計ジャッジ時間 17,682 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<vector>
#include<string>
#include<utility>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<functional>
#include<math.h>
#include<random>
#include <bitset>
using namespace std;
#define N (1000000000+7)
//#define N 998244353
#define INF 1e16
typedef long long ll;
typedef pair<ll,ll> P;
typedef pair<int,P> Q;
 
const int inf = (int)1e9; 
 
ll gcd(ll a, ll b) {
	if (b > a) {
		ll tmp = b;
		b = a;
		a = tmp;
	}
	if (a%b == 0)return b;
	else return gcd(b, a%b);
}

 
 
int main(void){
	ll n,d;
	cin>>n>>d;
	vector<int>v1,v2;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			v1.push_back(i*i+j*j);
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			if(d+i*i-j*j>=0 && d+i*i-j*j<=2*n*n){
				v2.push_back(d+i*i-j*j);
			}
		}
	}
	ll ans = 0;
	sort(v2.begin(),v2.end());
	for(int i=0;i<v1.size();i++){
		int index1 = upper_bound(v2.begin(),v2.end(),v1[i])-v2.begin();
		int index2 = lower_bound(v2.begin(),v2.end(),v1[i])-v2.begin();
		ans+=index1-index2;
	}
	cout<<ans<<endl;
	return 0;
}
0