[프로그래머스] 배열 조각하기

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

 

문제 설명이 조금 이해하기 힘들어서 생각보다 시간이 소요된 문제였다.

✅ query 배열을 순서대로 순회하면서, query[idx] 가 짝수인 경우 arr 배열에서 query[idx] 번째 인덱스 뒤의 요소들을 잘라서 버린다.

✅ query[idx] 가 홀수인 경우, arr 배열에서 query[idx] 번째 인덱스 앞의 요소들을 잘라서 버린다.

✅ 지문에서 'query[i] 번 인덱스를 제외하고~' 라는 부분이 있어서 헷갈렸는데, 이 뜻은 arr 에서 요소를 제거할 때 query[i] 번째 요소는 남겨두라는 의미였다! 🥲

✅ 파이썬의 슬라이스를 사용하면 쉽게 풀 수 있다.

 

📝  내가 작성한 코드

def solution(arr, query):
    for idx, qry in enumerate(query):
        if idx % 2 == 0:
            arr = arr[:qry+1]
        elif idx % 2 != 0:
            arr = arr[qry:]
            
    return arr