Проблема с подпоследовательностями арифметической прогрессии ⇐ Python

[Anonymous]

Я пытаюсь решить следующую задачу кода:

Написать программу, которая считывает два целых числа n и r , оба строго больше 1, за которыми следует последовательность целых чисел, и определяет, содержит ли последовательность последовательную подпоследовательность длины не менее n, которая образует арифметическую прогрессию с шагом r.
Последовательная подпоследовательность целых чисел образует арифметическую прогрессию с шагом r, если разница между двумя последовательными числами равна r. Например, 4 5 6 7 — арифметическая прогрессия с r = 1, а 22 33 44 55 66 — арифметическая прогрессия с r = 11.
Если входная последовательность содержит такую ​​последовательность, программа должна напечатать строку с первыми n элементами последовательности. В противном случае программа должна указать "Не найдено арифметической прогрессии с шагом r и длиной не менее n".
Ввод
Ввод состоит из двух целых чисел n>1 и r>1, за которыми следует последовательность целых чисел, содержащая как минимум 2 элемента.
Вывод
Если последовательность прогрессии имеет причину r и длина не менее n, выходные данные представляют собой первые n элементов последовательности. В противном случае на выходе будет "Не найдено арифметической прогрессии с шагом r и длиной не менее n".

Я пробовал с дрянная интерпретация предыдущей-текущей схемы, но тестер упражнений по-прежнему помечает свои частные тесты как неправильные ответы...
from yogi import scan

n = scan(int)
r = scan(int)
count = 1
found = False
nl = []
prev = -0.1
curr = scan(int)
while curr is not None:
if curr - prev == r:
nl.append(prev)
nl.append(curr)
count += 1
if count == n:
print(*sorted(set(nl)))
found = True
prev = curr
curr = scan(int)

if not found:
print(f"No arithmetic progression found with step {r} and length at least {n}")


Подробнее здесь: https://stackoverflow.com/questions/793 ... nces-issue