[프로그래머스] 2의 영역 - 파이썬

Programmers

2의 영역

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

프로그래머스

문제

코드

 

<풀이 1> → 실패(21.4 / 100)

def solution(arr):
    answer = []
    if 2 not in arr:
        answer = [-1]
    else:
        start_idx = arr[::-1].index(2)
        end_idx = len(arr) - arr[::-1].index(2)
        answer = arr[start_idx:end_idx]
    return answer

• arr[::-1] : 배열을 뒤집음
• arr[::-1].index(2) : 뒤집힌 배열에서 숫자 2가 처음으로 등장하는 인덱스
• len(arr) - arr[::-1].index(2) : 뒤집힌 배열의 숫자 2가 마지막으로 등장하는 인덱스

 

<풀이 2> → 성공

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

 

 

<풀이 3>

def solution(arr):
    answer = []
    lst=[]
    if 2 not in arr:
        return [-1]
    else:
        for i in range(0, len(arr)):
            if arr[i]==2:
                lst.append(i)
    return arr[lst[0]:lst[-1]+1]

 

Reference

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