結果

問題 No.800 四平方定理
ユーザー hint908
提出日時 2024-04-19 01:12:34
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,387 bytes
コンパイル時間 2,421 ms
コンパイル使用メモリ 196,240 KB
最終ジャッジ日時 2025-02-21 03:10:54
ジャッジサーバーID
（参考情報） 		judge3 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 3
other AC * 10 TLE * 1 -- * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using uint = unsigned int;
using ll = long long;
using ull = unsigned long long;
template<class T> using V = vector<T>;
template<class T> using VV = V<V<T>>;
template<class T> using VVV = V<VV<T>>;
template<class T> using VVVV = VV<VV<T>>;
#define rep(i,n) for(ll i=0ll;i<n;i++)
#define REP(i,a,n) for(ll i=a;i<n;i++)
const long long INF = (1LL << 60);
const long long mod99 = 998244353;
const long long mod107 = 1000000007;
const long long mod = mod99;
#define eb emplace_back
#define pb eb
#define be(v) (v).begin(),(v).end()
#define all(i,v) for(auto& i : v)
#define all2(i,j,v) for(auto& [i,j] : v)
template<class T, class U> void chmin(T& t, const U& u) { if (t > u) t = u; }
template<class T, class U> void chmax(T& t, const U& u) { if (t < u) t = u; }

// cin.tie(nullptr);
// ios::sync_with_stdio(false);
// cout << fixed << setprecision(20);

void solve(){
  ll n,d;
  cin >> n >> d;
  ll mm = n*n;
  ll cnt = 0;
  V<ll> v(4000001, 0);
  V<ll> w={1, 0, 3, 6};
  REP(i,1,n+1) v[i*i] = 1;
  REP(i,1,n+1) REP(j,i,n+1) REP(k,j,n+1){
  	ll a = i*i+j*j+k*k-d;
  	if(a <= 0) continue;
  	if(a > mm) break;
  	ll cc = 0;
  	if(i!=j) cc++;
  	if(j!=k) cc++;
  	if(i!=k) cc++;
  	cnt += v[a]*w[cc];
  }
  cout << cnt << endl;
}





int main(){
  cin.tie(nullptr);
  ios::sync_with_stdio(false);
  int t=1;
  // cin >> t;
  rep(i,t) solve();
}
0