結果

問題 No.800 四平方定理
ユーザー rnano
提出日時 2023-02-21 13:46:19
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,388 ms / 2,000 ms
コード長 1,372 bytes
コンパイル時間 1,705 ms
コンパイル使用メモリ 173,276 KB
実行使用メモリ 67,268 KB
最終ジャッジ日時 2024-07-22 01:22:08
合計ジャッジ時間 20,436 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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;

vll A, B;

int main() {
  ll N, D; cin >> N >> D;
  Rep(i, 1, N+1) {
    Rep(j, 1, N+1) {
      A.push_back(i*i + j*j);
      B.push_back(i*i - j*j);
    }
  }
  sort(all(B));
  ll ans = 0;
  fore(a, A) {
    // a + b = D
    auto itr1 = lower_bound(all(B), D-a);
    auto itr2 = lower_bound(all(B), D-a+1);
    ans += (itr2 - itr1);
  }
  cout << ans << endl;
}
0