結果
問題 | No.800 四平方定理 |
ユーザー |
|
提出日時 | 2019-03-17 22:39:57 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 564 ms / 2,000 ms |
コード長 | 1,167 bytes |
コンパイル時間 | 1,486 ms |
コンパイル使用メモリ | 172,216 KB |
実行使用メモリ | 43,360 KB |
最終ジャッジ日時 | 2024-07-08 00:52:45 |
合計ジャッジ時間 | 9,485 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
#include <bits/stdc++.h>using namespace std;using ll = long long;int main() {cin.tie(0);ios::sync_with_stdio(false);int n, d;cin >> n >> d;vector<int> v1;for (int x = 1; x <= n; x++) {for (int y = 1; y <= n; y++) {v1.push_back(x * x + y * y);}}sort(v1.begin(), v1.end());vector<int> v2;for (int w = 1; w <= n; w++) {int lim = min(n * n, w * w + d);for (int z = 1; z * z <= lim; z++) {v2.push_back(d + w * w - z * z);}}sort(v2.begin(), v2.end());int i = 0, j = 0;ll ans = 0;while (i < v1.size() && j < v2.size()) {while (i < v1.size() && j < v2.size()) {if (v1[i] == v2[j]) break;if (v1[i] < v2[j]) i++;else j++;}int ii = i, jj = j;while (i < v1.size()) {if (v1[i] != v1[ii]) break;i++;}while (j < v2.size()) {if (v2[j] != v2[jj]) break;j++;}ll tmp = i - ii;tmp *= j - jj;ans += tmp;}cout << ans << endl;return 0;}