Тройки с одинаковым шагом
Представь числа на линейке: нужно найти тройки, где каждое следующее число дальше ровно на один и тот же шаг. Дан строго возрастающий список nums и число diff. Тройка индексов i < j < k подходит, если nums[j] - nums[i] == diff и nums[k] - nums[j] == diff. Верни количество таких троек.
Примеры
Ввод
[0, 1, 4, 6, 7, 10], 3Вывод
2Ввод
[4, 5, 6, 7, 8, 9], 2Вывод
2Ввод
[1, 2, 3], 1Вывод
1Нужно реализовать
def arithmetic_triplets(nums: list[int], diff: int) -> int:
Теги
arraysethashmap
Войдите, чтобы видеть историю отправок
ВойтиВойдите, чтобы использовать AI Mentor
ВойтиВидимые тесты
1
Ввод
[0,1,4,6,7,10], 3Ожидаемый
22
Ввод
[4,5,6,7,8,9], 2Ожидаемый
23
Ввод
[1,2,3], 1Ожидаемый
14
Ввод
[1,2,3], 5Ожидаемый
05
Ввод
[0,3,6,9,12], 3Ожидаемый
36
Ввод
[10], 1Ожидаемый
0Фокус-радио
На паузе · SomaFM · Fluid