結果

問題 No.800 四平方定理
ユーザー occhan
提出日時 2019-03-17 23:19:22
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 773 bytes
コンパイル時間 1,857 ms
コンパイル使用メモリ 176,232 KB
実行使用メモリ 97,820 KB
最終ジャッジ日時 2024-07-08 02:40:57
合計ジャッジ時間 26,305 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 2
other AC * 1 WA * 19 TLE * 1 -- * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#define ll long long
typedef pair<ll,ll> P;
#define M 1000000007
int main(){
  ll n,m,c=0;
  cin>>n>>m;
  map<ll,ll> ma;
  for(ll i=1;i<=n;i++){
    for(ll j=i+1;j<=n;j++){
      ma[j*j-i*i]=j;
    }
  }
  for(ll i=1;i<=n;i++){
    for(ll j=i;j<=n;j++){
      ll k=m-i*i-j*j;
      if(k==0){
        if(i==j){
          c+=3*(n-j)+1;
        }else{
          c+=6*(n-j)+3;
        }
        c%=M;
      }else{
k=abs(k);
      if(ma[k]>=j){
        if(i==j){
          if(ma[k]==i){
            c++;
          }else{
            c+=3;
          }
        }else{
          if(ma[k]==i||ma[k]==j){
            c+=3;
          }else{
            c+=6;
          }
        }
        c%=M;
      }
      }
    }
  }
  cout<<c<<endl;
}
0