[Python] 프로그래머스 2의 영역

문제 설명

정수 배열 arr가 주어집니다. 배열 안의 2가 모두 포함된 가장 작은 연속된 부분 배열을 return 하는 solution 함수를 완성해 주세요.
단, arr에 2가 없는 경우 [-1]을 return 합니다.

 

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

프로그래머스

제한 사항

1 ≤ arr의 길이 ≤ 100,000
1 ≤ arr의 원소 ≤ 10

예제

arr	result
[1, 2, 1, 4, 5, 2, 9]	[2, 1, 4, 5, 2]
[1, 2, 1]	[2]
[1, 1, 1]	[-1]
[1, 2, 1, 2, 1, 10, 2, 1]	[2, 1, 2, 1, 10, 2]

문제 풀이

def solution(arr):
    tmp = [i for i in range(len(arr)) if arr[i] == 2]
    if len(tmp) == 0:
        return [-1]
    return [arr[idx] for idx in range(tmp[0], tmp[-1]+1)]

---

파이썬을 독학하시는 분들에게 도움이 되길 바라며,
혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.