結果

問題 No.800 四平方定理
ユーザー beet
提出日時 2019-03-18 11:33:09
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 920 ms / 2,000 ms
コード長 588 bytes
コンパイル時間 1,774 ms
コンパイル使用メモリ 201,512 KB
最終ジャッジ日時 2025-01-06 23:24:26
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using Int = long long;
template<typename T1,typename T2> inline void chmin(T1 &a,T2 b){if(a>b) a=b;}
template<typename T1,typename T2> inline void chmax(T1 &a,T2 b){if(a<b) a=b;}


//INSERT ABOVE HERE
signed main(){
  Int n,d;
  cin>>n>>d;
  unordered_map<Int, Int> cnt;
  cnt.reserve(2*n*n);
  
  for(Int i=1;i<=n;i++)
    for(Int j=1;j<=n;j++)
      cnt[i*i+j*j]++;
  
  Int ans=0;
  for(Int i=1;i<=n;i++){
    for(Int j=1;j<=n;j++){
      Int v=i*i-j*j+d;
      if(v>0&&cnt.count(v)) ans+=cnt[v];
    }
  }
  cout<<ans<<endl;
  return 0;
}
0