코딩테스트/프로그래머스

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

Developer D 2024. 1. 6. 20:21

문제 설명
정수 배열 arr와 query가 주어집니다.
query를 순회하면서 다음 작업을 반복합니다.
짝수 인덱스에서는 arr에서 query[i]번 인덱스를 제외하고 배열의 query[i]번 인덱스 뒷부분을 잘라서 버립니다.
홀수 인덱스에서는 arr에서 query[i]번 인덱스는 제외하고 배열의 query[i]번 인덱스 앞부분을 잘라서 버립니다.
위 작업을 마친 후 남은 arr의 부분 배열을 return 하는 solution 함수를 완성해 주세요.

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

    return answer

 

반응형