結果

問題 No.1318 ABCD quadruplets
コンテスト
ユーザー minato
提出日時 2023-01-26 23:46:11
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 153 ms / 2,000 ms
コード長 1,626 bytes
コンパイル時間 1,832 ms
コンパイル使用メモリ 195,108 KB
最終ジャッジ日時 2025-02-10 07:08:27
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define rep(i, n) for (ll i = 0; i < (n); i++)
constexpr char ln = '\n';
int calc(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 ans[160001];
int main() {
    int N, M;
    cin >> N >> M;

    rep(a, M + 1) {
        int val = calc(a, a, a, a);
        if (val > N) break;
        ans[val]++;
    }

    rep(a, M + 1) {
        for (int b = a + 1; b <= M; b++) {
            int val1 = calc(a, a, a, b);
            if (val1 > N) break;
            ans[val1] += 4;
            int val2 = calc(a, a, b, b);
            if (val2 <= N) ans[val2] += 6;
            int val3 = calc(a, b, b, b);
            if (val3 <= N) ans[val3] += 4;
        }
    }

    rep(a, M + 1) {
        for (int b = a + 1; b <= M; b++) {
            for (int c = b + 1; c <= M; c++) {
                int val1 = calc(a, a, b, c);
                if (val1 > N) break;
                ans[val1] += 12;
                int val2 = calc(a, b, b, c);
                if (val2 <= N) ans[val2] += 12;
                int val3 = calc(a, b, c, c);
                if (val3 <= N) ans[val3] += 12;
            }
        }
    }

    rep(a, M + 1) {
        for (int b = a + 1; b <= M; b++) {
            for (int c = b + 1; c <= M; c++) {
                for (int d = c + 1; d <= M; d++) {
                    int val = calc(a, b, c, d);
                    if (val > N) break;
                    ans[val] += 24;
                }
            }
        }
    }

    rep(i, N + 1) cout << ans[i] << ln;
}
0