結果

問題 No.800 四平方定理
ユーザー jdoie
提出日時 2019-03-17 23:09:05
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 1,729 bytes
コンパイル時間 1,495 ms
コンパイル使用メモリ 165,160 KB
実行使用メモリ 21,548 KB
最終ジャッジ日時 2024-07-08 01:59:36
合計ジャッジ時間 16,921 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 2
other AC * 2 WA * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

//
//#define _GLIBCXX_DEBUG
//#define NDEBUG
#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
const ll linf = 1e18;
const int inf = 1e9;

#define vl vector<ll>
#define pg priority_queue<ll, vector<ll>, greater<ll>>
#define pq priority_queue<ll>
#define rp(i, a, n) for (int i = a; i < n; i++)
#define pr(i, a, n) for (int i = a; i >= n; i--)
#define fm(i, mp) for (auto i = (mp).begin(); i != (mp).end(); ++i)
#define al(v) (v).begin(), (v).end()
#define mz(x) memset(x, 0, sizeof(x))
#define mi(x) memset(x, 0x3F, sizeof(x))
#define ce(x) cout << (x) << endl
#define cs(x) cout << (x) << ' '
#define co(x) cout << (x)
#define cd(x, y) cout << setprecision(x) << (y) << endl
#define ma make_pair
#define pb push_back
#define fs first
#define sc second
#define ceo cout << endl
#define cso cout << ' '
clock_t startT, endT;
int time()
{
    endT = clock();
    cerr << (endT - startT) / 1000 << "ms" << endl;
    return (endT - startT) / 1000;
}

int main()
{
    startT = clock();
    cin.tie(0);
    ios::sync_with_stdio(0);
    //----------------------------
    ll n, d;
    cin >> n >> d;
    vl wz;
    rp(i, 1, n + 1)
    {
        rp(j, 1, n + 1)
        {
            if (i - j >= 0)
                wz.pb((i + j) * (i - j));
        }
    }
    sort(al(wz));
  //  rp(i, 0, wz.size()) cs(wz[i]);

    ll ans = 0;
    rp(i, 0, n + 1)
    {
        rp(j, 1, n + 1)
        {
            ll k = i * i + j * j - d;
            ll k2 = lower_bound(al(wz), k) - wz.begin();
            ll k3 = upper_bound(al(wz), k) - wz.begin();
            //    cs(k2);
            //  ce(k3);
            ans += k3 - k2;
        }
    }
    ce(ans);

    //----------------------------

    time();
    return 0;
}
//
0