문제 설명
정수 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
입출력 예
| n | slicer | num_list | resullt |
| 3 | [1,5,2] | [1, 2, 3, 4, 5, 6, 7, 8, 9] | [2, 3, 4, 5, 6] |
| 4 | [1,5,2] | [1, 2, 3, 4, 5, 6, 7, 8, 9] | [2, 4, 6] |
입출력 예 설명
입출력 예 #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]입니다.
C++ 코딩테스트 리스트 자르기 코드
#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;
}
댓글