Khi ~K~ lẻ, một bộ ba ~(a, b, c)~ thỏa mãn khi tất cả đều chia hết cho ~K~ nên số cách chọn trong trường hợp này là ~(\frac{n}{k})^3~ (xuất hiện mũ ~3~ vì ta cần tính cả số cách chọn của bộ ba).

Khi ~K~ chẵn, một bộ ba ~(a, b, c)~ thỏa mãn khi tất cả đều chia hết cho ~K~, hoặc tất cả đều đồng dư ~\frac{K}{2}~ mod ~K~. Đếm các bộ ba như thế bằng cách đếm số các số bé hơn ~N~ mod ~K = 0~ hoặc ~\frac{k}{2}~. Vì vậy, số cách chọn trong trường hợp này là ~(\frac{n + \frac{k}{2}}{k})^3 + (\frac{n} {k}) ^ 3~.

Code mẫu

#include <bits/stdc++.h>
using namespace std;

int main() {
    int n, k;
    cin >> n >> k;
    int d = n / k;
    if (k%2) {
        cout << 1LL * d * d * d;
    } else {
        int d2 = (n + k/2) / k;
        cout << 1LL * d * d * d + 1LL * d2 * d2 * d2;
    }
}