알고리즘

[프로그래머스] 코딩테스트 연습 - 5명씩(JAVA)

inthyes 2023. 7. 9. 20:05

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

문제

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


소스코드

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

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

 

소스코드 설명

1. a를 배열로 선언한다.

2. for문을 5단위로 반복하도록하고 이 때 a배열에 names[i]를 삽입한다.