結果

問題 No.800 四平方定理
ユーザー zero_kpr
提出日時 2019-03-18 00:28:59
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 211 ms / 2,000 ms
コード長 648 bytes
コンパイル時間 1,635 ms
コンパイル使用メモリ 170,472 KB
実行使用メモリ 128,256 KB
最終ジャッジ日時 2024-07-08 07:46:43
合計ジャッジ時間 5,245 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int main()
{
    ll N,D; cin >> N >> D;
    vector<ll> cnt1(2*N*N + 1),cnt2(2*N*N + 1);
    for(int x = 1; x < N + 1; x++)
    {
        for(int y = 1; y < N + 1; y++)
        {
            cnt1.at(x*x + y*y)++;
        }
    }
    for(int w = 1; w <= N; w++)
    {
        for(int z = 1; z <= N; z++)
        {
            if(w*w + D - z*z >= 0 && w*w + D - z*z <= 2*N*N)
            {
                cnt2.at(w*w + D - z*z)++;
        }   }
    }
    int ans = 0;
    for(int i = 1; i < 2*N*N + 1; i++)
    {
        ans += cnt1.at(i)*cnt2.at(i);
    }
    cout << ans << endl;
}
0