結果

問題 No.800 四平方定理
ユーザー hanbei_dayohanbei_dayo
提出日時 2020-08-11 19:52:16
言語 C++14
(gcc 12.3.0 + boost 1.83.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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 9 ms
5,248 KB
testcase_01 AC 16 ms
5,248 KB
testcase_02 AC 12 ms
5,248 KB
testcase_03 AC 13 ms
5,248 KB
testcase_04 AC 12 ms
5,248 KB
testcase_05 AC 14 ms
5,248 KB
testcase_06 AC 19 ms
5,248 KB
testcase_07 AC 14 ms
5,248 KB
testcase_08 AC 3 ms
5,248 KB
testcase_09 AC 16 ms
5,248 KB
testcase_10 AC 567 ms
23,244 KB
testcase_11 AC 599 ms
25,680 KB
testcase_12 AC 642 ms
26,192 KB
testcase_13 AC 495 ms
19,568 KB
testcase_14 AC 540 ms
24,392 KB
testcase_15 AC 641 ms
26,320 KB
testcase_16 AC 545 ms
24,696 KB
testcase_17 AC 546 ms
24,656 KB
testcase_18 AC 728 ms
27,728 KB
testcase_19 AC 716 ms
27,596 KB
testcase_20 AC 2 ms
5,248 KB
testcase_21 AC 2 ms
5,248 KB
testcase_22 AC 732 ms
27,596 KB
testcase_23 AC 1,251 ms
43,464 KB
testcase_24 AC 1,024 ms
34,892 KB
testcase_25 AC 1,246 ms
43,340 KB
testcase_26 AC 2 ms
5,248 KB
testcase_27 AC 2 ms
5,248 KB
testcase_28 AC 1,023 ms
34,764 KB
testcase_29 AC 1,160 ms
43,336 KB
testcase_30 AC 1,014 ms
34,764 KB
testcase_31 AC 945 ms
33,100 KB
testcase_32 AC 1,073 ms
43,468 KB
権限があれば一括ダウンロードができます

ソースコード

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