[프로그래머스 / C++] n개 간격의 원소들

문제

n개 간격의 원소들 : 문제 링크

문제 분석

• 정수 리스트 num_list와 정수 n이 주어질 때, num_list의 첫 번째 원소부터 마지막 원소까지 n개 간격으로 저장되어있는 원소들을 차례로 담은 리스트를 return
• num_list의 첫 번째 원소부터 저장해야 하므로 for loop의 초기값을 0으로 설정하고, 증감식에서 index가 n만큼 커지도록 함. 따라서 마지막 원소까지 n개 간격으로 answer 벡터에 저장하고 return

풀이

#include <vector>

using namespace std;

vector<int> solution(vector<int> num_list, int n) {
    vector<int> answer;
    
    for(int i = 0; i < num_list.size(); i += n) answer.push_back(num_list[i]);
    return answer;
}