수포자

문제 설명

수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 각 수포자들은 자신만의 찍기 패턴을 가지고 있습니다.

1번 수포자의 찍기 패턴: 1, 2, 3, 4, 5, 1, 2, 3, 4, 5, ...
2번 수포자의 찍기 패턴: 2, 1, 2, 3, 2, 4, 2, 5, 2, 1, 2, 3, 2, 4, 2, 5, ...
3번 수포자의 찍기 패턴: 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, ...

1번 문제부터 마지막 문제까지의 정답이 순서대로 들은 배열 answers가 주어졌을 때, 가장 많은 문제를 맞힌 사람이 누구인지 배열에 담아서 반환하는 solution 함수를 작성해야 합니다.

제한사항

• 시험은 최대 10,000 문제로 구성되어 있습니다.
• 문제의 정답은 1, 2, 3, 4, 5 중 하나입니다.
• 가장 높은 점수를 받은 사람이 여럿일 경우, 오름차순으로 정렬하여 반환해야 합니다.

입출력 예

입력	출력
[1,2,3,4,5]	[1]
[1,3,2,4,2]	[1,2,3]

입출력 예 설명

• 입출력 예 #1: 수포자 1은 모든 문제를 맞혔고, 수포자 2와 3은 모든 문제를 틀렸습니다. 따라서 가장 많은 문제를 맞힌 사람은 수포자 1입니다.
• 입출력 예 #2: 모든 사람이 2문제씩을 맞혔습니다. 수포자 1, 2, 3이 모두 가장 많은 문제를 맞혀서 반환할 때는 오름차순으로 정렬하여 [1, 2, 3]을 반환합니다.

코드

Solution.java

import java.util.ArrayList;
import java.util.Collections;

class Solution {
    public int[] solution(int k, int[] score) {
        int[] answer = new int[score.length];
        ArrayList<Integer> temp = new ArrayList<>();

        for(int i = 0; i < score.length; i++) {
            if(temp.size() < k) {
                temp.add(score[i]);
                Collections.sort(temp);
                answer[i] = temp.get(0);
                continue;
            }

            if(temp.size() == k) {
                int minNum = temp.get(0);

                if(minNum < score[i]) {
                    temp.remove(0);
                    temp.add(score[i]);
                    Collections.sort(temp);
                }

                answer[i] = temp.get(0);
            }
        }
        return answer;
    }
}

코드 설명

• 수포자들의 찍기 패턴을 각각 pattern1, pattern2, pattern3 배열에 저장합니다.
• scores 배열을 선언하여 각 수포자들의 점수를 저장할 것입니다. 초기 값은 모두 0으로 초기화됩니다.
• answers 배열을 반복하면서, 각 수포자들이 해당 위치에서 맞춘 경우 점수를 증가시킵니다.
• 가장 높은 점수를 계산하기 위해 Math.max 함수를 사용합니다.
• winners라는 ArrayList에 가장 높은 점수를 받은 수포자들의 번호를 저장합니다. 이 때, 수포자 번호는 1부터 시작하므로 +1을 해서 저장합니다.
• winners 리스트를 배열로 변환하여 결과로 반환합니다.