Быстро создавайте массивы с включениями

Генератор списка создаёт список из цикла в одном выражении. Это короче, быстрее и является стандартным приёмом на соревнованиях для преобразования данных.

Шаблон выглядит так: [expr for item in iterable]. Выражение слева выполняется один раз для каждого элемента, который выдаёт цикл.

Самая распространённая строка на соревнованиях: разделите входные данные и преобразуйте каждый токен. Этот приём за один понятный шаг преобразует текст в целые числа.

map(int, ...) применяет int к каждому токену. Оберните результат в list(...), чтобы материализовать значения для многократного использования.

Добавьте в конце условие if, чтобы оставить только подходящие элементы. Здесь при построении остаются только чётные числа.

Можно одновременно преобразовывать и фильтровать: изменяйте сохраняемые элементы слева, а отбирайте их с помощью завершающего if.

Нужен массив фиксированного размера? [0] * n мгновенно создаёт n нулей — идеально для счётчиков и таблиц DP.

Для двумерной сетки используйте включение. [[0] * c for _ in range(r)] создаёт независимые строки, а [[0] * c] * r использует одну общую строку.

Если нужны ещё и позиции, используйте в цикле enumerate: на каждом шаге Вы получаете и индекс, и значение.

Объедините чтение и цикл, чтобы сразу получить r строк чисел и построить всю сетку всего в двух коротких строках.

Такой же синтаксис создаёт множества и словари: используйте фигурные скобки. {x for x in nums} удаляет дубликаты, а пары key:value создают отображение.

Выберите конструкцию, которая считывает строку целых чисел, разделённых пробелами.

Теперь Вы умеете в одной строке создавать, фильтровать и изменять форму массивов, а также быстро считывать целые матрицы. Включения делают разбор ввода коротким и защищённым от ошибок. 💡