結果

問題 No.800 四平方定理
ユーザー mine691mine691
提出日時 2019-03-17 22:49:03
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 343 ms / 2,000 ms
コード長 710 bytes
コンパイル時間 1,379 ms
コンパイル使用メモリ 168,416 KB
実行使用メモリ 323,688 KB
最終ジャッジ日時 2024-07-08 01:14:20
合計ジャッジ時間 7,027 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const ll mod = 1e9 + 7;
const ll inf = (1 << 30) - 1;
const ll infll = (1LL << 61) - 1;

ll n, d, ans;

int main()
{
    cin >> n >> d;
    vector<ll> X(4 * n * n), Y(d + 6 * n * n);
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= n; j++)
        {
            X[i * i + j * j]++;
        }
    }
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= n; j++)
        {
            Y[i * i - j * j + 3 * n * n]++;
        }
    }
    for (ll i = 1; i <= 2 * n * n; i++)
    {
        ans += X[i] * Y[d - i + 3 * n * n];
        //printf("X:%d Y:%d i:%d\n", X[i], Y[d - i], i);
    }
    cout << ans << endl;
}
0