結果

問題 No.800 四平方定理
コンテスト
ユーザー beet
提出日時 2019-03-18 11:30:48
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,113 ms / 2,000 ms
コード長 687 bytes
コンパイル時間 1,957 ms
コンパイル使用メモリ 220,244 KB
最終ジャッジ日時 2025-01-06 23:23:47
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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;}


#include <ext/pb_ds/assoc_container.hpp>
using namespace __gnu_pbds;
template<typename T,typename U> 
using gmap = cc_hash_table<T, U, hash<T> >;

//INSERT ABOVE HERE
signed main(){
  Int n,d;
  cin>>n>>d;
  gmap<Int, Int> cnt;
  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) ans+=cnt[v];
    }
  }
  cout<<ans<<endl;
  return 0;
}
0