結果

問題 No.800 四平方定理
ユーザー tonegawatonegawa
提出日時 2020-08-17 10:48:02
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 1,231 bytes
コンパイル時間 3,859 ms
コンパイル使用メモリ 146,396 KB
最終ジャッジ日時 2025-01-13 02:18:05
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 RE * 2
other AC * 4 RE * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <string>
#include <vector>
#include <queue>
#include <deque>
#include <algorithm>
#include <set>
#include <map>
#include <bitset>
#include <cmath>
#include <functional>
#include <iomanip>
#define vll vector<ll>
#define vvvl vector<vvl>
#define vvl vector<vector<ll>>
#define VV(a, b, c, d) vector<vector<d>>(a, vector<d>(b, c))
#define VVV(a, b, c, d) vector<vvl>(a, vvl(b, vll (c, d)));
#define re(c, b) for(ll c=0;c<b;c++)
#define rep(a,b,c) for(ll a=b;a<c;a++)
#define all(obj) (obj).begin(), (obj).end()
typedef long long int ll;
typedef long double ld;
using namespace std;

void get(vll &a){re(i,a.size()) scanf("%lld",&a[i]);}
void get(vvl &a){re(i,a.size()) re(j,a[0].size()) scanf("%lld",&a[i][j]);}
void print(vll &a){re(i,a.size()) cout<<a[i]<<(i==a.size()-1?"\n":" ");}
void print(vvl &a){re(i,a.size()) re(j,a[0].size())cout<<a[i][j]<<(j==a[0].size()-1?"\n":" ");}

int main() {
  ll n, d;std::cin >> n >> d;
  vll pl(10000000, 0);
  vll mi(10000000, 0);

  for(ll i=1;i<=n;i++){
    for(ll j=1;j<=n;j++){
      ll x = i*i+j*j;
      pl[x]++;
      ll y = i*i-j*j+d;
      mi[y]++;
    }
  }

  ll ans = 0;
  for(int i=0;i<10000000;i++) ans += pl[i] * mi[i];
  std::cout << ans << '\n';
}
0