Основное
 
 
Отправитель Задача Язык Вердикт Время Память Отослано Протест.  
214208454 Дорешивание:
TravellerAK
1609C - 48 PyPy 3-64 Полное решение 374 мс 39244 КБ 2023-07-17 15:55:03 2023-07-17 15:55:03
→ Исходный код
n = 10 ** 6;prime = [True for i in range(n + 1)];prime[1] = False;p = 2
while p * p <= n:
	if prime[p]:
		for i in range(p * p, n + 1, p):prime[i] = False
	p += 1
for _ in range(int(input())):
    n, b = map(int, input().split());li = list(map(int, input().split()));ans = 0
    for i in range(n):
        if prime[li[i]]:
            j, c = i + b, 0
            while j < n and li[j] == 1:j, c = j + b, c + 1
            d, j = 0, i - b
            while j >= 0 and li[j] == 1:j, d = j - b, d + 1
            ans += c + d * (c + 1)
    print(ans)
?
Время: ? ms, память: ? КБ
Вердикт: ?
Ввод
?
Вывод участника
?
Ответ жюри
?
Комментарий чекера
?
Диагностика
?
Показать детали тестирования