結果

問題 No.800 四平方定理
ユーザー granddaifuku
提出日時 2020-04-25 17:47:05
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 91 ms / 2,000 ms
コード長 1,100 bytes
コンパイル時間 1,488 ms
コンパイル使用メモリ 168,540 KB
実行使用メモリ 73,632 KB
最終ジャッジ日時 2024-11-07 11:03:33
合計ジャッジ時間 4,118 ms
ジャッジサーバーID
（参考情報） 		judge4 / 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 FOR(i, a, b) for(int i = a; i < (int)b; ++i)
#define rrep(i, n) for(int i = ((int)n - 1); i >= 0; --i)

typedef long long ll;
typedef long double ld;

const int Inf = 1e9;
const double EPS = 1e-9;
const int MOD = 1e9 + 7;
const int N = 9000001;

int main() {
    cin.tie(nullptr);
    ios::sync_with_stdio(0);
    int n, d;
    cin >> n >> d;
    ll res = 0;
    vector<int> plus(N), minus(N);
    FOR (i, 1, n + 1) {
        FOR (j, 1, n + 1) {
            int x = i * i - j * j;
            if (x >= 0) {
                plus[x]++;
            } else {
                x = abs(x);
                minus[x]++;
            }
        }
    }
    FOR (i, 1, n + 1) {
        FOR (j, 1, n + 1) {
            int tmp = i * i + j * j;
            int rest = d - tmp;
            if (rest >= 0) {
                res += plus[rest];
            } else {
                rest = abs(rest);
                res += minus[rest];
            }
        }
    }
    cout << res << endl;
    
    return 0;
}
0