結果

問題 No.800 四平方定理
ユーザー _____TAB__________TAB_____
提出日時 2019-03-17 22:02:42
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 94 ms / 2,000 ms
コード長 490 bytes
コンパイル時間 604 ms
コンパイル使用メモリ 68,872 KB
実行使用メモリ 38,528 KB
最終ジャッジ日時 2024-07-07 22:54:16
合計ジャッジ時間 2,870 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
using namespace std;
vector<int> dp(9e6,0);

int main(){
  int N, D;
  cin >> N >> D;
  int ans = 0;
  int m = 4e6;
  for(int x = 1; x <= N; ++x){
    for(int w = 1; w <= N; ++w){
      ++dp[x*x-w*w+m];
    }
  }
  for(int y = 1; y <= N; ++y){
    for(int z = 1; z <= N; ++z){
      //if(y*y + z*z > D) break;
      int d = D - y*y - z*z;
      if(d+m >= (int)dp.size() or d+m < 0) continue;
      ans += dp.at(d+m);
    }
  }
  cout << ans << endl;
}
0