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

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

 

풀이

파이썬으로 풀었던 문제를 자바로 다시 풀어봤다.

 

패턴 3가지에 대해서 연산을 통해 맞힌 숫자를 계산했고, 패턴 중 가장 많이 맞은 개수를 max로 저장했다.

 

그리고 max와 같은 값을 갖는 것들의 인덱스를 저장한다.

같은 경우 오름차순으로 출력해야 하기 때문에 정렬, 그리고 배열로 출력한다.

 

전체 소스코드

import java.util.*;

class Solution {
    public int[] solution(int[] answers) {
        int[][] patterns = {
            {1,2,3,4,5},
            {2,1,2,3,2,4,2,5},
            {3,3,1,1,2,2,4,4,5,5}
        };
        
        int[] score = new int[3];
        int max = 0;
        for (int k = 0; k < patterns.length; k++) {
            int cnt = 0;
            int size = patterns[k].length;
            for (int i = 0; i < answers.length; i++) {
                if (answers[i] == patterns[k][i % size]) cnt++;
            }
            score[k] = cnt;
            max = Math.max(max, cnt);
        }
        
        
        List<Integer> list = new ArrayList<>();
        for (int i = 0; i < score.length; i++) {
            if (score[i] == max) list.add(i+1);
        }
        
        Collections.sort(list);
        
        int[] answer = new int[list.size()];
        for (int i = 0; i < list.size(); i++) {
            answer[i] = list.get(i);
        }
        
        return answer;
    }
}