[프로그래머스 / C++] 리스트 자르기

문제

리스트 자르기 : 문제 링크

문제 분석

• 정수 n과 정수 3개가 담긴 리스트 slicer 그리고 정수 여러 개가 담긴 리스트 num_list가 주어진다. slicer에 담긴 정수를 차례대로 a, b, c라고 할 때, n에 따라 다음과 같이 num_list를 슬라이싱 하려고 한다.

  • n = 1 : num_list의 0번 인덱스부터 b번 인덱스까지
  • n = 2 : num_list의 a번 인덱스부터 마지막 인덱스까지
  • n = 3 : num_list의 a번 인덱스부터 b번 인덱스까지
  • n = 4 : num_list의 a번 인덱스부터 b번 인덱스까지 c 간격으로

• 슬라이싱한 리스트를 저장할 정수형 벡터 answer을 초기화. if ~ else if문을 통해 n이 1부터 4일경우 각각을 수행. 각 경우에 맞게 초기화식, 조건식, 증가식을 설정하여 for loop를 수행하여 num_list를 슬라이싱하여 answer에 저장. 최종적으로 저장된 answer을 return

풀이

#include <vector>

using namespace std;

vector<int> solution(int n, vector<int> slicer, vector<int> num_list) {
    vector<int> answer;
    
    if(n == 1) {
        for(int i = 0; i <= slicer[1]; ++i) answer.push_back(num_list[i]);
    }
    else if(n == 2) {
        for(int i = slicer[0]; i < num_list.size(); ++i) answer.push_back(num_list[i]);
    }
    else if(n == 3) {
        for(int i = slicer[0]; i <= slicer[1]; ++i) answer.push_back(num_list[i]);
    }
    else if(n == 4) {
        for(int i = slicer[0]; i <= slicer[1]; i += slicer[2]) answer.push_back(num_list[i]);
    }
    return answer;
}