結果
問題 | No.800 四平方定理 |
ユーザー |
![]() |
提出日時 | 2019-03-18 00:04:00 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 220 ms / 2,000 ms |
コード長 | 1,172 bytes |
コンパイル時間 | 810 ms |
コンパイル使用メモリ | 96,140 KB |
実行使用メモリ | 128,256 KB |
最終ジャッジ日時 | 2024-07-08 07:41:58 |
合計ジャッジ時間 | 4,599 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
#include <cstdio>#include <cstdlib>#include <cmath>#include <climits>#include <cassert>#include <iostream>#include <iomanip>#include <string>#include <stack>#include <queue>#include <vector>#include <map>#include <set>#include <algorithm>#include <numeric>#include <bitset>#define all(c) c.begin(), c.end()#define rall(c) c.rbegin(), c.rend()#define debug(x) cerr << #x << ": " << x << endlusing namespace std;typedef long long ll;typedef pair<ll, ll> Pll;typedef pair<int, int> Pii;const ll MOD = 1000000007;const long double EPS = 1e-10;const int dyx[4][2] = {{ 0, 1}, {-1, 0}, {0,-1}, {1, 0}};int main() {ll n, d;cin >> n >> d;vector<ll> cnt1(2*n*n+1, 0), cnt2(2*n*n+1, 0);for(int i=1;i<=n;++i) {for(int j=1;j<=n;++j) {++cnt1[i*i+j*j];++cnt2[i*i-j*j + n*n];}}ll ans = 0LL;for(int i=1;i<=2*n*n;++i) {if(d-i+n*n < 0 || 2*n*n < d-i+n*n) continue;ans += (cnt1[i] * cnt2[d-i+n*n]) % MOD;ans %= MOD;// cerr << i << ": " << cnt1[i] << " " << d-i << ": " << cnt2[d-i+n*n] << endl;}cout << ans << endl;}