문제 출처

코딩테스트 연습 > 완전탐색 > 모의고사
https://school.programmers.co.kr/learn/courses/30/lessons/42840

문제 설명

1번 수포자의 패턴은 {1,2,3,4,5} 의 반복
2번 수포자의 패턴은 {2,1,2,3,2,4,2,5} 의 반복
3번 수포자의 패턴은 {3,3,1,1,2,2,4,4,5,5} 의 반복

모의고사 수학 문제 정답으로 주어지는 배열 answers가 있을 때, 가장 많이 맞춘 수포자(or 수포자들)을 배열에 넣어 return 하라.

접근 방법

% 수포자의 패턴을 가져오는 방식이 조금 이상함에 주의
각각 수포자 패턴을 정답의 요소들과 비교하여 각각 count한다.
count 한 것을 비교하여 가장 많이 맞춘 수포자(or 수포자들)을 answer 배열에 삽입한다.

import java.util.*;

class Solution {
    public int[] solution(int[] answers) {
        int count_1 = 0;
        int count_2 = 0;
        int count_3 = 0;


        int[] pattern_1 = {1, 2, 3, 4, 5};
        int[] pattern_2 = {2, 1, 2, 3, 2, 4, 2, 5};
        int[] pattern_3 = {3, 3, 1, 1, 2, 2, 4, 4, 5, 5};


        Queue<Integer> q_1 = new LinkedList<>();
        Queue<Integer> q_2 = new LinkedList<>();
        Queue<Integer> q_3 = new LinkedList<>();

        for (int p : pattern_1) q_1.offer(p);
        for (int p : pattern_2) q_2.offer(p);
        for (int p : pattern_3) q_3.offer(p);


        for (int a : answers) {
            if (q_1.peek() == a) count_1++;
            q_1.offer(q_1.poll());

            if (q_2.peek() == a) count_2++;
            q_2.offer(q_2.poll());

            if (q_3.peek() == a) count_3++;
            q_3.offer(q_3.poll());
        }


        int[] scores = {count_1, count_2, count_3};
        int maxScore = Math.max(scores[0], Math.max(scores[1], scores[2]));


        List<Integer> result = new ArrayList<>();
        for (int i = 0; i < scores.length; i++) {
            if (scores[i] == maxScore) {
                result.add(i + 1); 
            }
        }

        int index = 0;

        int answer[] = new int[result.size()];
        for(int k : result){
            answer[index++] = k;
        }
        return answer;
    }
}

알게된 점

해당 문제 풀기 직전에 PriorityQueue 문제를 풀고와서 그런지 pattern을 가져오는 방식을 Queue로 풀었다.
간단하게 index % pattern.length로 풀면 더 짧고 쉽다.

문제푼 흔적