[프로그래머스 코딩테스트] 5명씩(Java)

  문제 설명

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

  제한사항

• 5 <= names의 길이 <= 30
• 1 <= names의 원소의 길이 <= 10
• names의 원소는 영어 알파벳 소문자로만 이루어져 있습니다.

 

  입출력 예

names	result
["nami", "ahri", "jayce", "garen", "ivern", "vex", "jinx"]	["nami", "vex"]

class Solution {
    public String[] solution(String[] names) {
        String[] answer;
        if (names.length % 5 == 0) {
            answer = new String[names.length / 5];
        } else {
            answer = new String[names.length / 5 + 1];
        }
        
        int index = 0;
        for (int i = 0; i < names.length; i += 5) {
            answer[index++] = names[i];
            if (i + 5 > names.length) {
                break;
            }
        }
        return answer;
    }
}

  answer 배열을 생성합니다. 만약 names 배열의 길이를 5로 나눈 나머지가 0이라면, 즉 맞아떨어진다면 answer 배열의 크기는 names 배열의 길이를 5로 나눈 값으로 지정합니다. 그렇지 않다면 answer 배열의 크기를 names 배열의 길이를 5로 나눈 값에 1을 더해줍니다.
  answer에 할당하는 원소의 위치가 될 index 변수를 생성합니다. for문이 names의 길이만큼 순회하며 5씩 증가합니다. answer의 index번째 위치에 names의 i번째 원소를 할당합니다. 만약 i + 5가 names의 길이보다 크다면 break로 for문을 빠져나옵니다. 값이 증가하다보면 매개 변수 값에 따라서 names의 길이를 초과할 수 있습니다. 이럴 경우 오류가 발생합니다. 조건문을 이용하여 오류가 발생하지 않도록 방지하는 것입니다.