Подсчитайте тройки (a, b, c), такие что a + b, b + c и a + c делятся на K

Даны два целых числа «N» и «K», задача состоит в том, чтобы подсчитать количество троек (a, b, c) натуральных чисел, не превышающих «N», таких, что «a + b», «b + c», и «с + а» кратны «К». Обратите внимание, что «a», «b» и «c» могут совпадать, а могут и не совпадать в триплете.

Примеры:

Input: N = 2, K = 2 
Output: 2
Explanation: All possible triplets are  (1, 1, 1) and (2, 2, 2)

Input: N = 3, K = 2 
Output: 9  

Подход:

Запустите три вложенных цикла от «1» до «N» и проверьте, делятся ли i+j, j+l и l+i на «K». Увеличивайте счетчик, если условие истинно.

Ниже приведена реализация вышеуказанного подхода:

Анализ сложности:

• Временная сложность: O(n ³ ), так как три цикла вложены друг в друга
• Вспомогательное пространство: O (1), поскольку дополнительный массив не используется, поэтому используется постоянное пространство.