Как решить задачу на python 3?

Question

[Firesieht]

У нас имеется задача:
После рабочего дня, вечером, в лаборатории все обычно играют в свою любимую игру AICRAFT. Несмотря на бдительность ребят, боты-вирусы время от времени появляются в игре. Кто-то из стажеров был очень наблюдателен и обратил внимание остальных на то, что бот в чате появляется с определенной периодичностью. Ребята пометили каждое сообщение вируса символом “@”, теперь для того, чтобы избавиться от вируса навсегда, надо только посчитать периодичность его выхода в чат и запрограммировать удаление сообщения с таким периодом заблаговременно. Помогите ребятам посчитать период появления вируса в чате.

НА ВХОД:
N — количество строк в диалоге.

Далее на каждой строчке, подаются сообщения из чата (сообщение ВИРУС-БОТА заканчивается на знак “@”).

НА ВЫХОД:
Одно число — периодичность появления бота.

ПРИМЕР:
Вход:

7
Это хорошо
Ага
АгаАгаАгаАгаАгаАга@
АААУжс
Блин, это же бот
Блин, это же ботБлин, это же ботБлин, это же бот@
Грустинка

Выход:
3

У меня получилось такое решение

a = int(input())
counter = 0
fl = False
for i in range(a):
    if input()[-1] == '@':
        if fl: break
        fl = not fl
    if fl:
        counter += 1
print(counter)

По моим тестам и по данным к задаче примерам программа работает, но как только отправляешь ее на проверку сразу бракуется на 3 тесте, помогите исправить или найти ошибку.

Answer 1

[tsarevfs]

Код похож на правду. Можно придумать несколько крайних ситуаций в которых по условию не понятно что делать.
1. Если сообщений от бота 0 или 1
2. Могут ли быть сообщения не от бота с @ на конце?

Comments

[Firesieht]

На сколько я понял , что сообщений 0 быть не может, т.к это что то типр отрывок из чата где общаються игроки, а если одно, то тоже не понятно, а вот на счет того, что @ на конце, ьам вроже написано , что ребята вычислили сообщения бота и типо пометили их, так что не может

Answer 2

[Griboks]

N/(messages+'\n').count('@\n')

Comments

[Firesieht]

я не совсем понимаю, куда вставлять или вместо чего вставлять данную строку, можете рассказать?

[Griboks]

Firesieht,

НА ВХОД:
N — количество строк в диалоге.

Далее на каждой строчке, подаются сообщения из чата (сообщение ВИРУС-БОТА заканчивается на знак “@”).

НА ВЫХОД:
Одно число — периодичность появления бота.

def period(N:int, messages:str)->float:
      N/(messages+'\n').count('@\n')

Это функция, которая делает то, что вы описали.

[Firesieht]

@Griboks
Firesieht,

a = int(input())
counter = 0
def period(N:int, messages:str)->float:
      N/(messages+'\n').count('@\n')
print(counter)

должно получиться что-то типо такого? Я просто новичок и еще не разу не сталкивался с такими функциями и с функциями в python в целом

[Griboks]

Firesieht, я не знаю, что у вас должно получиться. Вы попросили функцию периода, я её написал. Функция банальная: делим количество строк на количество строк бота.

Answer 3

[Andrey Dugin]

Если не углубляться в детали:

indices = [index for index, message in enumerate(text.splitlines()) if message.rstrip().endswith('@')]
print(indices[1] - indices[0])