문제 설명
최대 5명씩 탑승가능한 놀이기구를 타기 위해 줄을 서있는 사람들의 이름이 담긴 문자열 리스트 names가 주어질 때, 앞에서 부터 5명씩 묶은 그룹의 가장 앞에 서있는 사람들의 이름을 담은 리스트를 return하도록 solution 함수를 완성해주세요. 마지막 그룹이 5명이 되지 않더라도 가장 앞에 있는 사람의 이름을 포함합니다.
제한사항
5 ≤ names의 길이 ≤ 30 1 ≤ names의 원소의 길이 ≤ 10 names의 원소는 영어 알파벳 소문자로만 이루어져 있습니다.
나의 코드
import java.util.List;
import java.util.ArrayList;
class Solution {
public String[] solution(String[] names) {
List<String> list = new ArrayList<>();
for(int i=0; i<names.length; i+=5) {
list.add(names[i]);
}
String[] answer = new String[list.size()];
for(int j=0; j<list.size(); j++) {
answer[j] = list.get(j);
}
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;
}
}
List
를 사용하지 않고 배열
로만 문제 해결
느낀 점
다른 사람 코드 중에 List
를 사용하지 않고 미리 배열의 길이를 정하여 배열로 바로 해결하는 방법도 있었다. 사고해야하는 건 배열로 푸는 게 더 복잡하지만, 코드로써는 보다 더 적은 메모리가 들어 효율적인 방법같다. 가독성/유지보수성 측면과 메모리 효율 측면에서 목적을 잘 정해 코드를 짜는 버릇을 해보자