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

 

조건 대로 짝수 인덱스 때에는 arr 리스트에서 query[i]번 인덱스 제외하고 뒷부분을 버리고 (앞부분만 사용)

홀수 인덱스 때에는 arr 리스트에서 query[i]번 인덱스는 제외하고 앞부분을 버리고 (뒷부분만 사용)

 

슬라이싱을 활용해서 짝수 인덱스면 앞부분만 사용하게 홀수 인덱스면 뒷부분만 사용하게 코드를 작성하였고,

 

변경한 것은 반복할 때 유지할 수 있게 arr = arr[ : query[i]+1] 이런 식으로 작성했다.

 

def solution(arr, query):
    for i in range(len(query)):
        if i%2==0:
            arr = arr[:query[i]+1] 
        elif i%2==1 :
            arr = arr[query[i]:] 
    return arr