結果

問題 No.800 四平方定理
ユーザー goodbatongoodbaton
提出日時 2019-03-17 22:05:22
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 33 ms / 2,000 ms
コード長 1,351 bytes
コンパイル時間 667 ms
コンパイル使用メモリ 97,440 KB
実行使用メモリ 34,264 KB
最終ジャッジ日時 2024-07-07 23:09:14
合計ジャッジ時間 1,918 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>

#include <iostream>
#include <complex>
#include <string>
#include <algorithm>
#include <numeric>
#include <vector>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <unordered_map>
#include <unordered_set>

#include <functional>
#include <cassert>

typedef long long ll;
using namespace std;

#ifndef LOCAL
#define debug(x) ;
#else
#define debug(x) cerr << __LINE__ << " : " << #x << " = " << (x) << endl;

template <typename T1, typename T2>
ostream &operator<<(ostream &out, const pair<T1, T2> &p) {
  out << "{" << p.first << ", " << p.second << "}";
  return out;
}

template <typename T>
ostream &operator<<(ostream &out, const vector<T> &v) {
  out << '{';
  for (const T &item : v) out << item << ", ";
  out << "\b\b}";
  return out;
}
#endif

#define mod 1000000007 //1e9+7(prime number)
#define INF 1000000000 //1e9
#define LLINF 2000000000000000000LL //2e18
#define SIZE 200010


int counter[2000*2000*2+1];

int main(){
  int n, d;

  cin >> n >> d;

  for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++)
      counter[i*i+j*j]++;

  ll ans = 0;

  for(int i=1;i<=n;i++){
    for(int j=1;j<=n;j++){
      int p = i*i + d - j*j;

      if(0 <= p && p <= 2000 * 2000 * 2)
        ans += counter[p];
    }
  }

  cout << ans << endl;

  return 0;
}
0