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

문제 :

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

프로그래머스

 

해결 :

import java.util.*;
class Solution {
    public int[] solution(int[] answers) { 
        
        int[] one = {1,2,3,4,5};
        int[] two = {2,1,2,3,2,4,2,5};
        int[] thr = {3,3,1,1,2,2,4,4,5,5};
        
        int[] score = {0,0,0};
        
        for(int i = 0 ; i<answers.length; i++){
            if(one[i%5] == answers[i]){
                score[0]++;    
            }
            
            if(two[i%8] == answers[i]){
                score[1]++;
            }
            
            if(thr[i%10] == answers[i]){
                score[2]++;
            }
            
        }
        
        int max = Math.max(score[0], Math.max(score[1], score[2]));
        
        List<Integer> ans = new ArrayList<Integer>();
        for(int i = 0; i<score.length ; i++) if(max == score[i]) ans.add(i+1);
        
        int[] dap = new int[ans.size()];
        for(int i = 0; i<ans.size(); i++) dap[i] = ans.get(i);
        return dap;
    }
}