[완전탐색] 프로그래머스 - 모의고사

programmers.co.kr/learn/courses/30/lessons/42840

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

 

내 풀이

import java.util.ArrayList;

class Solution {
    public int[] solution(int[] answers) {
        int[] answer = {};
        
        int[] m1 = { 1, 2, 3, 4, 5 };
        int[] m2 = { 2, 1, 2, 3, 2, 4, 2, 5 };
        int[] m3 = { 3, 3, 1, 1, 2, 2, 4, 4, 5, 5 };
        int res1 = 0, res2 = 0, res3 = 0;
        
        for( int i=0; i<answers.length; i++ ) {
            if( m1[i % m1.length] == answers[i] ) res1++;
            if( m2[i % m2.length] == answers[i] ) res2++;
            if( m3[i % m3.length] == answers[i] ) res3++;
        }
        
        int max = Math.max( res1, Math.max( res2, res3) );
        
        ArrayList<Integer> list = new ArrayList<Integer>();
        if( max == res1 ) list.add(1);
        if( max == res2 ) list.add(2);
        if( max == res3 ) list.add(3);
        
        answer = new int[ list.size() ];
        
        for( int i=0; i<answer.length; i++ ) {
            answer[i] = list.get(i);
        }
                
        
        return answer;
    }
}