結果

問題 No.800 四平方定理
コンテスト
ユーザー kakira9618
提出日時 2019-03-18 00:04:21
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 90 ms / 2,000 ms
コード長 588 bytes
コンパイル時間 444 ms
コンパイル使用メモリ 59,756 KB
実行使用メモリ 65,728 KB
最終ジャッジ日時 2024-07-08 07:42:03
合計ジャッジ時間 3,473 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
using namespace std;
using ll = long long;

int main() {
    int N, D;
    cin >> N >> D;
    vector<int> cnt1(2 * 2000 * 2000 + 1);
    vector<int> cnt2(2 * 2000 * 2000 + 1);
    for(int i = 1; i <= N; i++) {
        for(int j = 1; j <= N; j++) {
            cnt1[i * i + j * j]++;
            if (i * i - j * j + D >= 0 && i * i - j * j + D < cnt2.size()) cnt2[i * i - j * j + D]++;
        }
    }
    
    ll ans = 0;
    for(int i = 0; i < cnt1.size(); i++) {
        ans += cnt1[i] * cnt2[i];
    }
    cout << ans << endl;
    
    return 0;
}
0