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

 

 

 

* 문제 풀이

import java.util.*;

class Solution {
    public String[] solution(String[] names) {
        List<String> answer = new ArrayList<>();
        for(int i = 0; i < names.length; i++){
            if(i%5 == 0){
                answer.add(names[i]);
            }
        }
        return answer.toArray(new String[0]);
    }
}

ArrayList<String>을 생성해 answer를 저장.

for문을 통해 names 배열을 순회해 if 조건문에 해당하는 배열 요소를 찾는다.

배열 요소를 5로 나누어 나머지가 0일 경우, answer 리스트에 추가한다.

new String[0] : 길이가 0인 빈 문자열. String 타입의 배열을 만들지만 초기 크기가 0.

배열의 크기를 자동으로 조절해 알맞은 크기의 배열을 반환한다.

 

 

* 프로그래머스 다른 풀이

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

class Solution {
    public String[] solution(String[] names) {
        String[] answer = new String[(names.length-1)/5+1];
        for(int i=0; i<answer.length; i++) {
            answer[i] = names[5*i];
        }
        return answer;
    }
}