結果

問題 No.800 四平方定理
コンテスト
ユーザー joji
提出日時 2026-01-21 13:26:32
言語 C++23
(gcc 15.2.0 + boost 1.89.0)
結果
AC  
実行時間 72 ms / 2,000 ms
コード長 856 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 3,645 ms
コンパイル使用メモリ 334,384 KB
実行使用メモリ 34,464 KB
最終ジャッジ日時 2026-01-21 13:26:39
合計ジャッジ時間 5,411 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
using namespace std;

void solve() {
        int N, D;
        cin >> N >> D;
        vector<int> bucket(2 * N * N + 1, 0);
        for (int i = 1; i <= N; i++) {
                for (int j = 1; j <= N; j++) {
                        bucket[i * i + j * j]++;
                }
        }
        int64_t ans = 0;
        for (int i = 1; i <= N; i++) {
                for (int j = 1; j <= N; j++) {
                        int target = D + i * i - j * j;
                        if (target >= 2 && target <= 2 * N * N) {
                                ans += 1LL * bucket[target];
                        }
                }
        }
        cout << ans << '\n';
}

int main() {
        ios::sync_with_stdio(false);
        cin.tie(NULL);

        int t = 1;
        // cin >> t;
        while (t--) solve();

        return 0;
}
0