Почему он выдает ошибку времени выполнения на codeforces для python?

Question

Итак, я новичок в программировании и изучил основы python. Я проверяю свои навыки на codeforces, решая сначала некоторые из их более простых задач. Я пытаюсь сделать 158A на codeforces. Я думаю, что получил его, потому что он прошел несколько тестов, которые я назначил. Итак, я попытался отправить его, и он сказал мне кое-что об ошибке во время выполнения. Я действительно не знаю, что это такое, поэтому мне бы хотелось, чтобы кто-нибудь сказал мне, что это такое и как исправить это в моем коде. Вот ссылка на проблему: https://codeforces.com/problemset/problem/158/A

n = int(input())
k = int(input())
b = []
for i in range(n):
    a = int(input())
    b.append(a)
c = 0
for i in b:
    if i >= b[k]:
        c = c+1
    else:
        pass
print(c)

Answer 1

В задаче 158A ожидаемые входные данные:
1. Два числа (int), разделенных одним пробелом, где 1 ≤ k ≤ n ≤ 50
2. n целые числа, разделенные пробелами, где ai ≥ ai + 1

Существует также условие: Очки ДОЛЖНЫ быть положительный ( оценка> 0 ), чтобы вы могли продвинуться дальше
Это все, что вам нужно, я тестировал его и каждый раз получал ожидаемый результат

a = input("Input n and k: ")
n = int(a.split(" ")[0])
k = int(a.split(" ")[1])
b = input("Input  n scores: ")

willAdvance = 0
scores = b.split()
print(scores)
for element in scores:
    if int(element) >= int(scores[k-1]) and int(scores[k-1]) > 0:
        willAdvance += 1

print(willAdvance)

ТЕСТ

Ввод:

8 5
10 9 8 7 7 7 5 5

Выход:

6

Ввод:

4 6
0 0 0 0

Выход:

0

Answer 2

Введенный вами ввод - «8 5» и «10 9 8 7 7 7 5 5». Это не значит, что вам даны «8» и «5» как два разных входа. Это означает, что у вас очень длинная строка, содержащая числа, разделенные пробелами. Вы должны превратить их в список.

a = input()

n = int(a.split(" ")[0])
k = int(a.split(" ")[1])

a должно равняться «8 5». Затем мы превращаем строку в список, используя a.split(" "). Это даст ["8", "5"].