[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 간격으로

올바르게 슬라이싱한 리스트를 return하도록 solution 함수를 완성해주세요.

---

제한사항

• n 은 1, 2, 3, 4 중 하나입니다.
• slicer의 길이 = 3
• slicer에 담긴 정수를 차례대로 a, b, c라고 할 때
  • 0 ≤ a ≤ b ≤ num_list의 길이 - 1
  • 1 ≤ c ≤ 3
• 5 ≤ num_list의 길이 ≤ 30
• 0 ≤ num_list의 원소 ≤ 100

---

입출력 예

입출력 예 설명

입출력 예 #1

• [1, 2, 3, 4, 5, 6, 7, 8, 9]에서 1번 인덱스부터 5번 인덱스까지 자른 리스트는 [2, 3, 4, 5, 6]입니다.

입출력 예 #2

• [1, 2, 3, 4, 5, 6, 7, 8, 9]에서 1번 인덱스부터 5번 인덱스까지 2개 간격으로 자른 리스트는 [2, 4, 6]입니다.

---

작성코드

#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];
    
    switch (n)
    {
        case 1:
            answer = vector<int>(num_list.begin(), num_list.begin()+b+1);
            break;
        case 2:
            answer = vector<int>(num_list.begin()+a, num_list.end());
            break;
        case 3:
            answer = vector<int>(num_list.begin()+a, num_list.begin()+b+1);
            break;
        case 4:
            for (int i=a; i<=b; i+=c)
            {
                answer.push_back(num_list[i]);
            }
            break;
    }
    
    return answer;
}

---

문제 URL

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

프로그래머스

---

반응형

728x90