結果

問題 No.800 四平方定理
ユーザー T1610T1610
提出日時 2019-03-17 21:26:31
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 838 bytes
コンパイル時間 1,481 ms
コンパイル使用メモリ 175,948 KB
実行使用メモリ 81,568 KB
最終ジャッジ日時 2024-07-07 20:20:06
合計ジャッジ時間 22,516 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 20 TLE * 1 -- * 9
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
  
using namespace std;
  
#define rep(i,n) REP(i,0,n)
#define REP(i,s,e) for(int i=(s); i<(int)(e); i++)
#define repr(i, n) REPR(i, n, 0)
#define REPR(i, s, e) for(int i=(int)(s-1); i>=(int)(e); i--)
#define pb push_back
#define all(r) r.begin(),r.end()
#define rall(r) r.rbegin(),r.rend()
#define fi first
#define se second
  
typedef long long ll;
typedef vector<int> vi;
typedef vector<ll> vl;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
 
const int INF = 1e9;
const ll MOD = 1e9 + 7;
double EPS = 1e-8;

int main(){
    ll n, d;
    cin >> n >> d;
    map<ll, ll> mp;
    REP(x, 1, n+1) REP(y, 1, n+1) ++mp[(ll)x*x+(ll)y*y];
    ll ans = 0LL;
    REP(z, 1, n+1) REP(w, 1, n+1) {
        ll xy = (ll)w*w+d-ll(z*z);
        if(mp.count(xy)) ans+=mp[xy];
    }
    cout << ans << '\n';
    return 0;
}
0