結果

問題 No.800 四平方定理
ユーザー null_mnnull_mn
提出日時 2019-03-20 20:40:54
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,488 ms / 2,000 ms
コード長 660 bytes
コンパイル時間 1,801 ms
コンパイル使用メモリ 172,940 KB
実行使用メモリ 83,788 KB
最終ジャッジ日時 2024-09-19 00:50:10
合計ジャッジ時間 21,351 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long int;

int main() {
    ll n, d;
    cin >> n >> d;
    //A(:=xx+yy),B(:=ww-zz+D)の取り得る値を列挙
    vector<ll> A, B;
    for (ll x = 1; x <= n; ++x) {
        for (ll y = 1; y <= n; ++y) {
            A.push_back(x * x + y * y);
        }
    }
    for (ll x = 1; x <= n; ++x) {
        for (ll y = 1; y <= n; ++y) {
            B.push_back(x * x - y * y + d);
        }
    }
    sort(B.begin(), B.end());
    int cnt = 0;
    for (auto a : A) {
        cnt += upper_bound(B.begin(), B.end(), a) - lower_bound(B.begin(), B.end(), a);
    }
    cout << cnt << endl;
    return 0;
}
0