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

ybk·2023년 9월 23일
0

프로그래머스

목록 보기
56/236

Lv.0 배열 조각하기

문제 설명

정수 배열 arr와 query가 주어집니다. query를 순회하면서 다음 작업을 반복합니다.

  • 짝수 인덱스에서는 arr에서 query[i]번 인덱스를 제외하고 배열의 query[i]번 인덱스 뒷부분을 잘라서 버립니다.
  • 홀수 인덱스에서는 arr에서 query[i]번 인덱스는 제외하고 배열의 query[i]번 인덱스 앞부분을 잘라서 버립니다.
    위 작업을 마친 후 남은 arr의 부분 배열을 return 하는 solution 함수를 완성해 주세요.

입출력

코드 풀이

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

출처: 프로그래머스 코딩테스트 연습

profile
개발자 준비생~

0개의 댓글