結果

問題 No.800 四平方定理
コンテスト
ユーザー rnano
提出日時 2023-02-21 13:58:37
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 208 ms / 2,000 ms
コード長 1,286 bytes
コンパイル時間 1,679 ms
コンパイル使用メモリ 166,408 KB
実行使用メモリ 127,104 KB
最終ジャッジ日時 2024-07-22 01:31:29
合計ジャッジ時間 5,822 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;
#define rep(i,n) for (int i = 0; i < (int)(n); ++i)
#define rrep(i,n) for (int i = (int)(n-1); i >= 0; --i)
#define Rep(i,a,b) for (int i = a; i < b; ++i)
#define rRep(i,a,b) for (int i = a; i > b; --i)
#define fore(i,a) for(auto &i:a)
#define all(x) (x).begin(),(x).end()
#define rall(x) (x).rbegin(),(x).rend()
#define Bit(x,i) (((x)>>(i))&1)
using ll = long long;
using vi = vector<int>;
using vll = vector<long long>;
using vvi = vector<vi>;
using vvll = vector<vll>;
using vb = vector<bool>;
using vs = vector<string>;
using pii = pair<int, int>;
using pll = pair<ll, ll>;
template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; }
template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return 1; } return 0; }
struct settings{settings(){ios_base::sync_with_stdio(0); cin.tie(0); cout<<fixed<<setprecision(15);};}settingss;
const double PI = acos(-1);
const int INF = 1001001001;
const ll LINF = 1001001001001001001LL;

ll A[8000010], B[13000010];

int main() {
  ll N, D; cin >> N >> D;
  Rep(i, 1, N+1) {
    Rep(j, 1, N+1) {
      A[i*i + j*j]++;
      B[i*i - j*j + D + 2000*2000]++;
    }
  }
  ll ans = 0;
  rep(i, 8000005) {
    ans += A[i] * B[i + 2000*2000];
  }
  cout << ans << endl;
}
0