[Python/코테공부] 2의 영역

[Python/코테공부] 2의 영역

2의영역

 

내 풀이

def solution(arr):
    answer = [i for i, v in enumerate(arr) if v==2]
    return arr[answer[0]:answer[-1]+1] if len(answer)>0 else [-1]

 

테스트 결과

내 풀이 결과

 

다른 사람은 반복문을 사용하지 않고 풀어서 정리한다.

 

다른 사람의 풀이

def solution(arr):
    if 2 not in arr:
        return [-1]
    return arr[arr.index(2) : len(arr) - arr[::-1].index(2)]

 

테스트 결과

다른 사람 풀이 테스트 결과

 

확실히 인덱스 찾고 슬라이싱만 하니 속도가 더 빠르다.

 

* 문제 출처 프로그래머스

https://school.programmers.co.kr/learn/courses/30/lessons/181894

프로그래머스