[프로그래머스 LV.0] 배열 조각하기 — Python 정답 & 해설

🔗 1. 문제 URL

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

프로그래머스

 

 

---

 

📝 2. 문제 요구사항 정리

정수 배열 arr와 query가 주어졌을 때, query를 순서대로 확인하면서 arr의 특정 구간만 남기는 문제입니다. query의 인덱스가 짝수인지 홀수인지에 따라 잘라내는 방향이 달라집니다.

• 짝수 인덱스: query[i]번 인덱스까지는 남기고, 그 뒤쪽 원소들을 잘라냅니다.
• 홀수 인덱스: query[i]번 인덱스부터 끝까지 남기고, 그 앞쪽 원소들을 잘라냅니다.
• 최종 반환값: 모든 query를 처리한 뒤 남아 있는 arr의 부분 배열을 반환합니다.

 

---

 

💡 3. 핵심 개념

✔ 1) query의 인덱스 기준으로 동작 나누기

이 문제에서 중요한 점은 query의 값 자체가 아니라, 현재 query가 몇 번째 인덱스에 있는지입니다. 인덱스가 짝수라면 뒤쪽을 제거하고, 홀수라면 앞쪽을 제거해야 하므로 idx % 2 값을 기준으로 분기할 수 있습니다.

✔ 2) Python 슬라이싱으로 배열 자르기

Python에서는 슬라이싱을 이용하면 배열의 일부 구간만 간단하게 가져올 수 있습니다. arr[:val + 1]은 0번 인덱스부터 val번 인덱스까지 남기고, arr[val:]은 val번 인덱스부터 마지막 원소까지 남깁니다.

✔ 3) 매 단계마다 갱신되는 arr 사용하기

query를 처리할 때마다 arr의 길이와 원소 구성이 달라집니다. 따라서 처음 주어진 arr를 기준으로 계속 계산하는 것이 아니라, 매번 잘라낸 결과를 다시 arr에 저장하면서 다음 query를 처리해야 합니다.

 

---

 

✅ 4. 정답 Python 코드

def solution(arr, query):
    for idx, val in enumerate(query):
        if idx % 2 == 1: # 홀수 인덱스
            arr = arr[val : ]
        else: # 짝수 인덱스
            arr = arr[ : val + 1]
    
    return arr

 

---

 

💬 5. 풀이하면서 느낀 점

문제 설명만 보면 앞부분과 뒷부분을 번갈아 자르는 과정이 조금 헷갈릴 수 있지만, query의 인덱스가 짝수인지 홀수인지에 따라 동작을 나누면 훨씬 단순하게 정리할 수 있는 문제였습니다.

특히 Python 슬라이싱의 끝 인덱스는 포함되지 않는다는 점을 잘 기억해야 했습니다. 짝수 인덱스에서 query[i]번 인덱스까지 남겨야 하므로 arr[:val + 1]처럼 +1을 해주는 부분이 핵심이었습니다.

 

궁금하신 점이나 포스팅에 대한 피드백이 있다면 댓글로 편하게 남겨주세요.

도움이 되셨다면 공감이나 댓글 한 줄 남겨주시면 포스팅을 이어가는 데 큰 힘이 됩니다. 😊