[Level.0] 2의 영역

문제 설명: 정수 배열 arr가 주어집니다. 배열 안의 2가 모두 포함된 가장 작은 연속된 부분 배열을 return 하는 solution 함수를 완성해 주세요.

• 단, arr에 2가 없는 경우 [-1]을 return 합니다.

제한사항

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

문제 해결

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

 

주석: index(2)를 사용하여 첫 번째 2의 시작 지점을 확보하고 배열을 반전시켜 뒤에서부터 첫 번째 2의 위치를 계산해 마지막 인덱스를 추출함. 이를 통해 모든 2를 포함하는 최소 범위의 부분 배열을 슬라이싱하여 반환함.