컴프리헨션으로 배열 빠르게 만들기

리스트 컴프리헨션은 하나의 표현식으로 반복문에서 리스트를 만듭니다. 더 짧고 빠르며 데이터를 구성할 때 대회에서 표준적으로 사용됩니다.

형태는 [expr for item in iterable]입니다. 왼쪽의 표현식은 반복문이 반환하는 각 항목마다 한 번씩 실행됩니다.

대회에서 가장 흔한 줄은 입력을 split하고 각 토큰을 변환하는 형태입니다. 이렇게 하면 한 번의 읽기 쉬운 단계로 텍스트를 정수로 매핑할 수 있습니다.

map(int, ...)는 각 토큰에 int를 적용합니다. 값을 반복해서 사용하려면 list(...)로 감싸 실제 리스트로 만드십시오.

끝에 if 조건을 추가하면 조건을 통과한 항목만 남길 수 있습니다. 여기서는 짝수만 생성 결과에 남습니다.

한 번에 변환과 필터링을 수행할 수도 있습니다. 뒤에 붙인 if로 항목을 선별하면서 왼쪽의 항목을 변경합니다.

고정된 크기의 배열이 필요하신가요? [0] * n은 n개의 0을 즉시 만들어 주므로 카운터와 DP 테이블에 안성맞춤입니다.

2차원 격자에는 컴프리헨션을 사용하세요. [[0] * c for _ in range(r)]은 서로 독립적인 행을 만들지만, [[0] * c] * r은 하나의 행을 공유합니다.

위치도 필요할 때는 enumerate로 반복하세요. 그러면 각 단계에서 인덱스와 값을 함께 깔끔하게 얻을 수 있습니다.

입력과 반복을 결합하면 숫자로 된 r개의 행을 한 번에 읽어 전체 격자를 짧은 두 줄로 만들 수 있습니다.

같은 문법으로 집합과 딕셔너리도 만들 수 있습니다. 중괄호를 사용하세요. {x for x in nums}는 중복을 제거하고, key:value 쌍은 매핑을 만듭니다.

공백으로 구분된 정수 한 줄을 읽는 관용 표현을 골라 보세요.

이제 한 줄로 배열을 만들고, 필터링하고, 형태를 바꿀 수 있으며 전체 행렬도 빠르게 읽을 수 있습니다. 컴프리헨션을 사용하면 입력 처리가 짧아지고 오류도 줄어듭니다. 💡