結果

問題 No.800 四平方定理
コンテスト
ユーザー cotton_fn_
提出日時 2019-06-30 18:56:23
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,254 bytes
コンパイル時間 855 ms
コンパイル使用メモリ 109,236 KB
最終ジャッジ日時 2025-01-07 05:46:07
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 2
other AC * 28 WA * 1 RE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <deque>
#include <queue>
#include <array>
#include <set>
#include <map>
#include <cmath>
#include <algorithm>
#include <numeric>
#include <cassert>
#include <utility>
#include <tuple>
#include <functional>
#include <bitset>
#include <cstdint>

using namespace std;
using i64 = int64_t;
using i32 = int32_t;
template<class T, class U> void init_n(vector<T>& v, size_t n, U x) 
{ v = vector<T>(n, x); }
template<class T> void init_n(vector<T>& v, size_t n) { init_n(v, n, T()); }
template<class T> void read_n(vector<T>& v, size_t n, size_t o = 0) 
{ v = vector<T>(n+o); for (size_t i=o; i<n+o; ++i) cin >> v[i]; }
template<class T> void read_n(T a[], size_t n, size_t o = 0)
{ for (size_t i=o; i<n+o; ++i) cin >> a[i]; }
template<class T> T gabs(const T& x) { return max(x, -x); }
#define abs gabs

i64 n, d;
int main() {
  cin >> n >> d;
  vector<i64> a(2 * n * n + 1), b(2 * n * n + 1);
  for (i64 x = 1; x <= n; ++x) {
    for (i64 y = 1; y <= n; ++y) {
      a[x * x + y * y]++;
      i64 z = x * x - y * y + d;
      if (z > 0) b[z]++;
    }
  }
  i64 ans = 0;
  for (i64 i = 1; i <= a.size(); ++i) {
    ans += a[i] * b[i];
  }
  cout << ans << '\n';
  return 0;
}
0