結果

問題 No.800 四平方定理
ユーザー izryt(趣味)izryt(趣味)
提出日時 2020-05-24 09:23:56
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,269 bytes
コンパイル時間 4,587 ms
コンパイル使用メモリ 99,540 KB
最終ジャッジ日時 2025-01-10 15:28:35
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 TLE * 1
other AC * 22 TLE * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <cstdio>
#include <string>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <utility>
#include <functional>
#include <climits>
#include <cstring>

using namespace std;

//#define int long long
#define rep(i, n) for (int i=0;i<(int)(n);++i)
#define rep1(i, n) for (int i=1;i<=(int)(n);++i)
#define range(i, l, r) for (int i=l;i<(int)(r);++i)
#define rrange(i, l, r) for (int i=r-1;i>=(int)(l);--i)
#define unless(a) if(!(a))
#define all(a) begin(a),end(a)
#define fst first
#define scd second
#define PB emplace_back
#define PPB pop_back

using vi=vector<int>;
using pii=pair<int, int>;
using ll=long long;

bool chmin(int&a,int b){return a>b?(a=b,true):false;}
bool chmax(int&a,int b){return a<b?(a=b,true):false;}
//int read(){int a;scanf("%lld",&a);return a;}

const int inf = 1e9 + 10;
const int mod = 1e9 + 7;

int N, D;
int xy[8000800];
map<int, int> zw;

signed main()
{
	cin >> N >> D;

	int mx = 0;

	range(x, 1, N + 1) {
		range(y, 1, N + 1) {
			xy[x * x + y * y]++;
		}
	}

	range(z, 1, N + 1) {
		range(w, 1, N + 1) {
			zw[z * z - w * w]++;
		}
	}

	ll ans = 0;

	range(i, 1, 4000040) {
		if (zw.find(D - i) != zw.end())
			ans += xy[i] * zw[D - i];
	}

	printf("%lld\n", ans);
}
0