배열 조각하기
문제설명
정수 배열 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
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
'C++ > 코딩 기초 트레이닝' 카테고리의 다른 글
[C++/프로그래머스] 순서 바꾸기 (0) | 2023.12.11 |
---|---|
[C++/프로그래머스] n 번째 원소부터 (0) | 2023.12.11 |
[C++/프로그래머스] 2의 영역 (0) | 2023.10.25 |
[C++/프로그래머스] 배열 만들기 3 (0) | 2023.10.25 |
[C++/프로그래머스] 첫 번째로 나오는 음수 (0) | 2023.10.25 |