結果

問題 No.800 四平方定理
ユーザー mine691
提出日時 2019-03-17 22:47:26
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 747 bytes
コンパイル時間 1,866 ms
コンパイル使用メモリ 174,732 KB
実行使用メモリ 267,136 KB
最終ジャッジ日時 2024-07-08 01:09:27
合計ジャッジ時間 36,975 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 2 TLE * 1
other WA * 10 TLE * 11 -- * 9
権限があれば一括ダウンロードができます

ソースコード

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;
map<int, int> Y;

int main()
{
    cin >> n >> d;
    vector<int> X(3 * 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]++;
        }
    }
    for (int i = 1; i <= 2 * n * n; i++)
    {
        if (Y.find(d - i) != Y.end())
            continue;
        ans += X[i] * Y[d - i];
        //printf("X:%d Y:%d i:%d\n", X[i], Y[d - i], i);
    }
    cout << ans << endl;
}
0