프로그래머스 - 5명 씩

프로그래머스 - 5명 씩

import java.util.List;
import java.util.ArrayList;

class Solution {
    public String[] solution(String[] names) {
        List<String> list = new ArrayList<>();
       if(names.length % 5 == 0) {
           for(int i=0; i<names.length; i = i+5) {
               list.add(names[i]);
           }
       }
        
        else {
            int tmp = names.length / 5;
            // 맨 처음꺼 넣기(첫번째 그룹)
            list.add(names[0]);
            // 5, 10, 15, ... 인덱스가 그룹의 첫번째 사람들=> 나머지가 있더라도 처음 시작인덱스
            for(int i=1; i<=tmp; i++) {
                list.add(names[i*5]);
            } 
        }
        
        return list.toArray(new String[0]);
    }
}