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

문제 : https://programmers.co.kr/learn/courses/30/lessons/42840

코딩테스트 연습 - 모의고사 | 프로그래머스

 

딱히 잘 푼것 같지 않다. 그냥 풀었다

 

나의 코드 : 

import java.util.*;
 
 
class Solution {
    int [] p1;
    int [] p2;
    int [] p3;
    public int[] solution(int[] answers) {
        p1 = new int[]{1,2,3,4,5};//5
        p2 = new int[]{2,1,2,3,2,4,2,5};//8
        p3 = new int[]{3,3,1,1,2,2,4,4,5,5};//10
        
        int a1=0;
        int a2=0;
        int a3=0;
        
        for(int i=0; i<answers.length; i++){
            int tmp1 = i%5;
            if(answers[i] == p1[tmp1]) a1++;  
            
            int tmp2 = i%8;
            if(answers[i] == p2[tmp2]) a2++;
            
            int tmp3 = i%10;
            if(answers[i] == p3[tmp3])  a3++;
        }
        
        //find max
        
        int max = a1;
        if(a1<a2){
            max = a2;
        }
        if(max<a3){
            max = a3;
        }
        
        ArrayList<Integer> correct= new ArrayList<>();
        
        if(a1==max) correct.add(1);
        if(a2==max) correct.add(2);
        if(a3==max) correct.add(3);
            
        Collections.sort(correct);
        
        
    
        int [] answer = new int[correct.size()];
        
        for(int i=0; i<correct.size(); i++){
            answer[i] = correct.get(i);
        }
        return answer;
    }
}