結果
問題 |
No.1318 ABCD quadruplets
|
ユーザー |
![]() |
提出日時 | 2020-12-15 22:01:46 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 115 ms / 2,000 ms |
コード長 | 2,049 bytes |
コンパイル時間 | 1,954 ms |
コンパイル使用メモリ | 77,544 KB |
最終ジャッジ日時 | 2025-01-17 01:12:02 |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
#pragma GCC target("avx2") #pragma GCC optimize("O3") #pragma GCC optimize("unroll-loops") #include<iostream> using namespace std; inline int sum(int a, int b, int c, int d) { return a * a + a * b + a * c + a * d + b * b + b * c + b * d + c * c + c * d + d * d; } int main() { cin.tie(0); ios::sync_with_stdio(0); int n, m; cin >> n >> m; int cnt[160001] = {}; for (int i = 0, sa = 0; i <= m; ++i, sa += i * 2 - 1) { for (int j = 0, sb = sa; j < i; ++j, sb += i + j * 2 - 1) { for (int k = 0, sc = sb; k < j && sc <= n; ++k, sc += i + j + k * 2 - 1) { for (int l = 0, sd = sc; l < k && sd <= n; ++l, sd += i + j + k + l * 2 - 1) { cnt[sd] += 24; } } } } for (int i = 0, sa = 0; i <= m; ++i, sa += i * 2 - 1) { for (int j = 0, sb = sa; j < i; ++j, sb += i + j * 2 - 1) { for (int k = 0, sc = sb; k < j && sc <= n; ++k, sc += i + j + k * 2 - 1) { { int s = sc + (i + j + k + i) * i; if (s <= n) cnt[s] += 12; } { int s = sc + (i + j + k + j) * j; if (s <= n) cnt[s] += 12; } { int s = sc + (i + j + k + k) * k; if (s <= n) cnt[s] += 12; } } } } for (int i = 0; i <= m; ++i) { for (int j = 0; j < i; ++j) { { int s = sum(i, i, i, j); if (s <= n) cnt[s] += 4; } { int s = sum(i, j, j, j); if (s <= n) cnt[s] += 4; } { int s = sum(i, i, j, j); if (s <= n) cnt[s] += 6; } } } for (int i = 0; i <= m; ++i) { int s = sum(i, i, i, i); if (s <= n) cnt[s] += 1; } for (int i = 0; i <= n; ++i) { cout << cnt[i] << '\n'; } return 0; }