Crie Arrays Rapidamente com Compreensões

Uma compreensão de listas cria uma lista a partir de um laço em uma única expressão. Ela é mais curta, mais rápida e o padrão das competições para transformar dados.

O padrão é [expr for item in iterable]. A expressão à esquerda é executada uma vez para cada elemento produzido pelo laço.

A linha mais comum em competições: divida a entrada e converta cada valor. Essa etapa legível converte texto em inteiros.

map(int, ...) aplica int a cada valor. Envolva-o em list(...) para materializar os valores e reutilizá-los.

Adicione uma condição if no final para manter apenas os itens aprovados. Aqui, somente os números pares sobrevivem à construção.

É possível transformar e filtrar ao mesmo tempo: altere os itens mantidos à esquerda enquanto os seleciona com o if final.

Precisa de um vetor com tamanho fixo? [0] * n cria n zeros instantaneamente, sendo perfeito para contadores e tabelas de DP.

Use uma compreensão para uma grade 2D. [[0] * c for _ in range(r)] cria linhas independentes, enquanto [[0] * c] * r compartilha uma única linha.

Quando também precisar das posições, percorra os elementos com enumerate para obter o índice e o valor juntos em cada etapa.

Combine a leitura e o laço para obter de uma vez r linhas de números, construindo a grade completa em duas linhas curtas.

A mesma sintaxe cria conjuntos e dicionários: use chaves. {x for x in nums} remove duplicatas, e pares chave:valor criam um mapa.

Escolha a forma idiomática de ler uma linha de inteiros separados por espaços.

Agora você consegue construir, filtrar e remodelar vetores em uma linha, além de ler matrizes completas rapidamente. As compreensões mantêm o processamento da entrada curto e livre de erros. 💡