結果

問題 No.800 四平方定理
コンテスト
ユーザー ningenMe
提出日時 2019-03-17 23:23:28
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 3,117 bytes
コンパイル時間 995 ms
コンパイル使用メモリ 109,548 KB
実行使用メモリ 114,164 KB
最終ジャッジ日時 2024-07-08 02:57:13
合計ジャッジ時間 9,338 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 1
other AC * 3 WA * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <fstream>
#include <cmath>  
#include <cstdlib>
#include <ctime>
#include <algorithm>
#include <numeric>
#include <functional>
#include <string> 
#include <vector>
#include <bitset>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <deque>
#include <random>


using namespace std;
using ll = long long;
using ull = unsigned long long;

#define REP(i,n) for(long long i = 0; i < (n); i++)
#define FOR(i, m, n) for(long long i = (m);i < (n); ++i)
#define ALL(obj) (obj).begin(),(obj).end()

template<class T> using V = vector<T>;
template<class T, class U> using P = pair<T, U>;

const ll MOD = (ll)1e9 + 7;
const ll MOD2 = 998244353;
const ll LLINF = (ll)1e18;
const ll INTINF = (ll)1e9;
const long double PI = 3.1415926535897932384626433;

template <class T> void corner(bool flg, T hoge) { if (flg) { cout << hoge << endl; exit(0); } }
template <class T, class U>ostream &operator<<(ostream &o, const map<T, U>&obj) { o << "{"; for (auto &x : obj) o << " {" << x.first << " : " << x.second << "}" << ","; o << " }"; return o; }
template <class T>ostream &operator<<(ostream &o, const set<T>&obj) { o << "{"; for (auto itr = obj.begin(); itr != obj.end(); ++itr) o << (itr != obj.begin() ? ", " : "") << *itr; o << "}"; return o; }
template <class T>ostream &operator<<(ostream &o, const vector<T>&obj) { o << "{"; for (int i = 0; i < (int)obj.size(); ++i)o << (i > 0 ? ", " : "") << obj[i]; o << "}"; return o; }
template <class T, class U>ostream &operator<<(ostream &o, const pair<T, U>&obj) { o << "{" << obj.first << ", " << obj.second << "}"; return o; }
template <template <class tmp>  class T, class U> ostream &operator<<(ostream &o, const T<U> &obj) { o << "{"; for (auto itr = obj.begin(); itr != obj.end(); ++itr)o << (itr != obj.begin() ? ", " : "") << *itr; o << "}"; return o; }
void print(void) { cout << endl; }
template <class Head> void print(Head&& head) { cout << head; print(); }
template <class Head, class... Tail> void print(Head&& head, Tail&&... tail) { cout << head << " "; print(forward<Tail>(tail)...); }
template <class T> void chmax(T& a, const T b) { a = max<T>(a, b); }
template <class T> void chmin(T& a, const T b) { a = min<T>(a, b); }
void YN(bool flg) { cout << ((flg) ? "YES" : "NO") << endl; }
void Yn(bool flg) { cout << ((flg) ? "Yes" : "No") << endl; }
void yn(bool flg) { cout << ((flg) ? "yes" : "no") << endl; }

int main() {
	V<P<int, int>> c(8000000 + 1, { -1,-1 });
	ll N,D; cin >> N >> D;
	for(int i = 1; i <= N; ++i){
		for(int j = i; j <= N; ++j){
			c[i*i + j*j] = {i,j};
		}
	}

	map<V<int>,int> mp;
	ll ans = 0;
	for(int w = 1; w <= N; ++w){
		for (int i = 1; i <= N; ++i) {
			int z = i;
			int t = w*w - z*z + D;
			if (t > 0 && c[t].first != -1) {
				V<int> tmp = { c[t].first,c[t].second,z };
				sort(ALL(tmp));
				tmp.push_back(w);
				if (mp[tmp]>0) continue;
				mp[tmp]++;
				if (tmp[0] == tmp[2]) ans += 1;
				else if (tmp[0] == tmp[1] && tmp[1] != tmp[2]) ans += 3;
				else if (tmp[0] != tmp[1] && tmp[1] == tmp[2]) ans += 3;
				else ans += 6;
			}
		}
	}
	cout << ans << endl;
	return 0;
}
0