모의고사

import java.util.ArrayList;
import java.util.HashMap;
 
class Solution {
    public int[] solution(int[] answers) {
        int[][] person = { { 1, 2, 3, 4, 5 }, { 2, 1, 2, 3, 2, 4, 2, 5 }, { 3, 3, 1, 1, 2, 2, 4, 4, 5, 5 } };
        int[] answer = {};
 
        int n, i, j, max;
        HashMap<Integer, Integer> map = new HashMap<Integer, Integer>();
        ArrayList<Integer> num = new ArrayList<Integer>();
        
        for (i = 1; i <= 3; i++)
            map.put(i, 0);
        
        for (n = 0; n < answers.length; n++) {
            for (i = 0; i < 3; i++) {
                j = n % person[i].length;
                if (answers[n] == person[i][j]) {
                    map.put(i + 1, map.get(i + 1) + 1);
                }
            }
        }
        max = Math.max(Math.max(map.get(1), map.get(2)), map.get(3));
        
        if (max != 0) {
            for (i = 1; i <= 3; i++)
                if (max == map.get(i))
                    num.add(i);
 
            answer = new int[num.size()];
            for (i = 0; i < num.size(); i++)
                answer[i] = num.get(i);
        }
        return answer;
    }
}

hashmap의 keySet을 이용하여 정렬을 시도했지만 잘 되지 않아서 수정하였다.