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

꿈별·2023년 8월 25일
0

문제풀이

목록 보기
32/52

문제


풀이

#include <string>
#include <vector>

using namespace std;

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

    else if(3 == n)
    {
        for (int i = a; i <= b; ++i)
        {
            answer.push_back(num_list[i]);
        }
    }
   
    else
    {
        for (int i = a; i <= b; i+=c)
        {
            answer.push_back(num_list[i]);
        }
    }

    return answer;
}

0개의 댓글