Question

Пожалуйста, помогите решить задачу Кинотеатр!!

В кинотеатре n рядов по m мест в каждом. В двумерном массиве хранится информация о проданных билетах, число 1 означает, что билет на данное место уже продан, число 0 означает, что место свободно. Поступил запрос на продажу k билетов на соседние места в одном ряду. Определите, можно ли выполнить такой запрос.

Входные данные
Программа получает на вход числа n≤30 и m≤30. Далее идут n строк, содержащих m чисел (0 или 1), разделённых пробелами. Затем дано число k.

Выходные данные
Программа должна вывести номер ряда, в котором есть k подряд идущих свободных мест. Если таких рядов несколько, то выведите номер наименьшего подходящего ряда. Если подходящего ряда нет, выведите число 0.

Answer 1

#создаём матрицу, рассматриваем все возможные срезы длиной k

#и при первом подходящем наборе нулей записываем ответ

#Python

n, m = map(int, input().split())

hall = [list(map(int, input().split())) for i in range(n)]

k = int(input())

ans = 0

flag = True

for i in range(n):

   if flag:

       for j in range(m-k+1):

           if hall[i][j:j+k] == [0]*k:

               ans = i + 1

               flag = False

print(ans)