[프로그래머스] Lv.0 5명씩

📎 문제 출처

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

📌 문제 설명

최대 5명씩 탑승가능한 놀이기구를 타기 위해 줄을 서있는 사람들의 이름이 담긴 문자열 리스트 names가 주어질 때, 앞에서 부터 5명씩 묶은 그룹의 가장 앞에 서있는 사람들의 이름을 담은 리스트를 return하도록 solution 함수를 완성해주세요. 마지막 그룹이 5명이 되지 않더라도 가장 앞에 있는 사람의 이름을 포함합니다.

❓ 풀이 방법

반복문의 index 를 5씩 키우면서 돌린다.

📌 Code

class Solution {
    public String[] solution(String[] names) {
        int len = names.length;
        int N = len % 5 != 0 ? len / 5 + 1 : len / 5;
        String[] answer = new String[N];
        
        int index = 0;
        for(int i = 0; i < names.length; i += 5) {
            answer[index++] = names[i];
        }
        
        return answer;
    }
}