썩은 사과 찾기 해설
사과의 상태는 리스트 형태로 주어지며, 썩은 사과는 '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]
다음 내용이 궁금하다면?
코드프렌즈 PLUS 멤버십 가입 or 강의를 등록해 주세요!