[프로그래머스 (Java)] 모의고사

https://programmers.co.kr/learn/courses/30/lessons/42840?language=java 

코딩테스트 연습 - 모의고사

 

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

class Solution {
public Integer[] solution(int[] answers) {
        int[] score = {0, 0, 0};
    
        int[] student1 = {1, 2, 3, 4, 5}; 
        int[] student2 = {2, 1, 2, 3, 2, 4, 2, 5};
        int[] student3 = {3, 3, 1, 1, 2, 2, 4, 4, 5, 5};

        for (int i = 0; i < answers.length; i++) {
            int ans = answers[i];
            if (ans == student1[i % 5]) {
                score[0]++;
            }
            if (ans == student2[i % 8]) {
                score[1]++;
            }
            if (ans == student3[i % 10]) {
                score[2]++;
            }
        }

        List<Integer> res = new ArrayList<>();
        int max = Arrays.stream(score).max().getAsInt();
        for (int i = 0; i < score.length; i++) {
            if (score[i] == max) {
                res.add(i + 1);
            }
        }

        Integer[] answer = res.toArray(new Integer[res.size()]);
        Arrays.sort(answer);
        return answer;
    }
}