結果

問題 No.800 四平方定理
コンテスト
ユーザー NaruY
提出日時 2019-03-18 15:11:15
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 2,185 bytes
コンパイル時間 1,647 ms
コンパイル使用メモリ 166,160 KB
実行使用メモリ 68,940 KB
最終ジャッジ日時 2024-07-18 09:42:14
合計ジャッジ時間 12,269 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 RE * 1
other AC * 4 RE * 26
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function 'int main()':
main.cpp:76:24: warning: iteration 5000002 invokes undefined behavior [-Waggressive-loop-optimizations]
   76 |         ans += count1[i] * count2[i];
      |                ~~~~~~~~^
main.cpp:5:41: note: within this loop
    5 | #define FORq(i, m, n) for(int i = (m);i <= (n);++i)
      |                                         ^
main.cpp:75:5: note: in expansion of macro 'FORq'
   75 |     FORq(i,0,5000002){
      |     ^~~~

ソースコード

diff # 

#include <bits/stdc++.h>
#define FOR(i,a,b) for(int i=(a);i<(b);++i)
#define REP(i,n)  for(int i=0;i<(n);++i)
#define REPr(i,n) for(int i=(n)-1;i>=0; --i)
#define FORq(i, m, n) for(int i = (m);i <= (n);++i)
#define FORqr(i, m , n) for(int i = (n);i >=(m);--i)
#define SCD(n) scanf("%d",&n)
#define SCD2(m,n) scanf("%d%d",&m,&n)
#define SCD3(m,n,k) scanf("%d%d%d",&m,&n,&k)
#define SCLLD(n) scanf("%lld",&n)
#define SCLLD2(m,n) scanf("%lld%lld",&m,&n)
#define SCLLD3(m,n,k) scanf("%lld%lld%lld",&m,&n,&k)
#define PB push_back
#define MP make_pair
#define ARSCD(A,N) REP(i,N){SCD(A[i]);}
#define ARSCD1(A,N) FORq(i,1,N){SCD(A[i]);}
#define VSCD(v,N) REP(i,N){int (x); SCD(x); v.PB(x);}
#define VSCLLD(v,N) REP(i,N){long long (x); SCLLD(x); v.PB(x);}
#define PRINTD(n) printf("%d\n",n)
#define PRINTLLD(n) printf("%lld\n",n)
#define DEBUG printf("%s\n","debug")
#define fst first
#define snd second
#define SIN(x,S) (S.count(x) != 0)
#define M0(x) memset(x,0,sizeof(x))
#define FILL(x,y) memset(x,y,sizeof(x))
#define MM(x) memset(x,-1,sizeof(x))
#define ALL(x) (x).begin(),(x).end()
using namespace std;
typedef pair<int,int> PII;
typedef pair<long long,long long> PLL;
typedef vector<int> VI;
typedef vector < VI > VVI;
typedef vector<long long> VL;
typedef long long ll;
typedef long long integer;
///////////////////////////////////////////////
const ll MOD = 1000000007;
ll gcd(ll x,ll y){return y?gcd(y,x%y):x;}

template<class T> inline bool chmin(T& a, T b) {
    if (a > b) {
        a = b;
        return true;
    }
    return false;
}
template<class T> inline bool chmax(T& a, T b) {
    if (a < b) {
        a = b;
        return true;
    }
    return false;
}
///////////////////////////////////////////////
int main(){
   static ll count1[5000002]={};
   static ll count2[5000002]={};
   int N; cin >> N;
   ll D; cin >> D;

   for(ll x = 1; x <= N; x++){
       for(ll y = 1; y <= N ; y++){
           count1[x*x+y*y]++;
       }
   }

    for(ll z = 1; z <= N; z++){
        for(ll w = 1; w <= N ; w++){
            count2[w*w-z*z+D]++;
        }
    }

    ll ans = 0;
    FORq(i,0,5000002){
        ans += count1[i] * count2[i];
    }

    cout << ans << "\n";

}
0