💻 자바 알고리즘 - n개 간격의 원소들

📕 문제 설명

정수 리스트 num_list와 정수 n이 주어질 때, num_list의 첫 번째 원소부터 마지막 원소까지 n개 간격으로 저장되어있는 원소들을 차례로 담은 리스트를 return하도록 solution 함수를 완성해주세요.

 

📕 제한 사항

• 5 ≤ num_list의 길이 ≤ 20
• 1 ≤ num_list의 원소 ≤ 9
• 1 ≤ n ≤ 4

 

📝 답안

class Solution {
    public int[] solution(int[] num_list, int n) {
        int count = (num_list.length % n == 0)?num_list.length/n : (num_list.length/n)+1;
        int[] answer = new int[count];

        for(int i=0; i<count; i++) {
            answer[i] = num_list[i*n];
        }
        return answer;
    }
}

num_list 배열의 원소 중 n개 간격에 저장되어 있는 원소들만 저장한 배열을 만들기 위해선 우선 배열의 길이를 구해야한다. 배열의 길이는 num_list를 n개로 나눴을 때 나머지가 0인 경우는 그 나눈 몫이 길이가 되고 만약 나머지가 있는 경우는 몫에 +1을 해준다.

 

그렇게 만든 배열 길이 값만큼 for문을 돌면서 answer에 num_list 원소를 넣어주는데 이 때 num_list 인덱스 값으로 i*n을 넣어서 n개 간격에 있는 원소를 가져온다.