結果

問題 No.800 四平方定理
コンテスト
ユーザー Y17
提出日時 2019-03-17 22:21:14
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 1,562 ms / 2,000 ms
コード長 739 bytes
コンパイル時間 1,624 ms
コンパイル使用メモリ 165,112 KB
実行使用メモリ 89,640 KB
最終ジャッジ日時 2024-07-08 00:05:54
合計ジャッジ時間 17,112 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

int main(){
    long long n, d;
    cin >> n >> d;

    long long ans = 0;
    unordered_map<long long, long long> mp;

  	long long mi = LLONG_MAX;
  	long long ma = 0;
    for(long long i = 1;i <= n;i++){
        for(long long j = 1;j <= n;j++){
            if(i*i-j*j+d > 0 && i*i-j*j+d <= n*n+n*n){
                mp[i*i-j*j+d]++;
              	ma = max(ma, i*i-j*j+d);
              	mi = min(mi, i*i-j*j+d);
            }
        }
    }

    for(long long i = 1;i <= n;i++){
        for(long long j = 1;j <= n;j++){
          	if(i*i+j*j < mi) continue;
          	if(i*i+j*j > ma) break;
            ans += mp[i*i+j*j];
        }
    }


    cout << ans << endl;

    return 0;
}
0