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

배열 조각하기

---

문제설명

정수 배열 arr와 query가 주어집니다.

query를 순회하면서 다음 작업을 반복합니다.

• 짝수 인덱스에서는 arr에서 query[i]번 인덱스를 제외하고 배열의 query[i]번 인덱스 뒷부분을 잘라서 버립니다.
• 홀수 인덱스에서는 arr에서 query[i]번 인덱스는 제외하고 배열의 query[i]번 인덱스 앞부분을 잘라서 버립니다.

위 작업을 마친 후 남은 arr의 부분 배열을 return 하는 solution 함수를 완성해 주세요.

---

제한사항

• 5 ≤ arr의 길이 ≤ 100,000
  • 0 ≤ arr의 원소 ≤ 100
• 1 ≤ query의 길이 < min(50, arr의 길이 / 2)
  • query의 각 원소는 0보다 크거나 같고 남아있는 arr의 길이 보다 작습니다.

---

입출력 예

---

작성코드

#include <string>
#include <vector>

using namespace std;

vector<int> solution(vector<int> arr, vector<int> query) {
    vector<int> answer;
    
    for(int i=0; i<query.size(); i++)
    {
        if (i % 2 ==0) // 짝수 인덱스라면
        {
            // query[i]번 인덱스 뒷부분 잘라내기
            if(query[i]+1 < arr.size())
            {
                arr.erase(arr.begin() + query[i] + 1, arr.end());
            }
        }
        else
        {
            // 홀수에서는 query[i]번 인덱스 앞부분 잘라냄
            if(query[i] > 0)
            {
                arr.erase(arr.begin(), arr.begin() + query[i]);
            }
        }
    }
    answer = arr;
    return answer;
}

---

문제 URL

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

프로그래머스

---

반응형

728x90