結果
問題 | No.800 四平方定理 |
ユーザー |
![]() |
提出日時 | 2020-06-13 19:55:21 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 73 ms / 2,000 ms |
コード長 | 858 bytes |
コンパイル時間 | 638 ms |
コンパイル使用メモリ | 73,472 KB |
最終ジャッジ日時 | 2025-01-11 03:57:24 |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
#include <iostream> #include <vector> #include <map> using namespace std; using ll = long long; void ins() {} template<class T,class... Rest>void ins(T& v,Rest&... rest){cin>>v;ins(rest...);} #define _overload3(_1,_2,_3,name,...) name #define _rep2(i,n) for(int i=0,_i=(n);i<_i;++i) #define _rep3(i,a,b) for(int i=(a),_i=(b),_ii=(b)>(a)?1:-1;abs(_i-i)>0;i+=_ii) #define rep(...) _overload3(__VA_ARGS__,_rep3,_rep2)(__VA_ARGS__) constexpr int MAX_N = 3000; int m[MAX_N*MAX_N+1], m2[MAX_N*MAX_N+1]; int main() { int N, D; ins(N, D); rep(i, MAX_N*MAX_N+1) m[i] = m2[i] = 0; m2[0] += N; rep(i, N) { ++m[2*(i+1)*(i+1)]; rep(j, i+1, N) { m[(i+1)*(i+1)+(j+1)*(j+1)] += 2; ++m2[(j+1)*(j+1)-(i+1)*(i+1)]; } } ll ans = 0; rep(i, MAX_N*MAX_N+1) { ans += m2[abs(i-D)] * m[i]; } cout << ans << endl; return 0; }