썩은 사과 찾기 해설

사과의 상태는 리스트 형태로 주어지며, 썩은 사과는 '0', 신선한 사과는 '1'로 표시됩니다.

함수는 이 리스트를 입력 받아 썩은 사과의 위치(인덱스)를 찾아 리스트 형태로 반환합니다.

복사
def solution(apples):
    return [index for index, apple in enumerate(apples) if apple == 0]

• enumerate(apples)를 사용하여 각 사과의 인덱스와 값을 가져옵니다.

• if apple == 0을 통해 썩은 사과를 필터링합니다.

• 리스트 컴프리헨션을 사용하여 썩은 사과의 인덱스를 모두 반환합니다.

복사
print(solution([1, 0, 1, 0, 1]))  # 출력: [1, 3]