Задание 17 ЕГЭ по информатике

Question

В файле содержится закольцованная последовательность целых чисел, по модулю не превышающих 100000. Определите количество троек элементов последовательности, в которых только одно число не является четырехзначными, а сумма элементов тройки является полным квадратом. В ответе запишите два числа: сначала количество найденных троек, затем минимальную из сумм элементов таких троек. В данной задаче под тройкой подразумевается три различных элемента последовательности, находящихся на равном друг от друга расстоянии.
Файл к заданию: https://storage.yandexcloud.net/100points-bank/informatics-ege/files/10004_17.txt

Accepted Answer

def f(x): return 1000 <= abs(x) < 10000
a = list(map(int, open('17.txt')))
d = len(a) // 3
b = []
for x, y, z in zip(a, a[d:], a[d*2:]):
s = x + y + z
if f(x) + f(y) + f(z) == 2 and s >= 0 and s ** 0.5 % 1 == 0:
b += [s]
print(len(b), min(b))

Задание 17 ЕГЭ по информатике

Задания на обработка целочисленных данных и проверку делимости