[Programmers] 모의고사 - Java

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

프로그래머스

---

1. 문제

수포자 3명이 수학문제를 찍는다.

문제 수는 최대 10000개이고 각자 찍는 패턴이 있다.

1번 수포자가 찍는 방식: 1, 2, 3, 4, 5, 1, 2, 3, 4, 5, ...
2번 수포자가 찍는 방식: 2, 1, 2, 3, 2, 4, 2, 5, 2, 1, 2, 3, 2, 4, 2, 5, ...
3번 수포자가 찍는 방식: 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, ...

정답 배열 answers가 주어졌을 때 가장 많은 문제를 맞힌 수포자의 번호를 배열 형태로 반환한다.

가장 많은 문제를 맞춘 수포자가 여럿일 경우 오름차순으로 반환한다.

 

2. 풀이

완전탐색 문제이다.

어렵지 않은 문제라고 생각하여 신나게 짜서 제출했는데 5, 7 테스트케이스에서 계속 막혔다.

2시간을 고민하다 원인을 찾지 못하여 결국 블로그를 슬쩍 참고하게 되었다..

원인은 Math.max에 있었다.

 

나는 처음에 아래와 같이 max 값을 구해주었다.

max = Math.max(p1_cnt, p2_cnt);
max = Math.max(p2_cnt, p3_cnt);
max = Math.max(p3_cnt, p1_cnt);

 

하지만 정답 코드는 아래와 같았다.

max = Math.max(p1_cnt, Math.max(p2_cnt, p3_cnt));

 

왜인지 방금 깨달았다. 지금 생각해보니 이걸 문제로 여기지 않았다는게 놀랍다

max는 한 줄로 비교해줘야 한다. 이렇게 해야 p1_cnt, p2_cnt, p3_cnt 3개 값을 다같이 비교한다.

반면에 수정 전 코드는 p1_cnt, p2_cnt 비교 후 끝이다. 줄마다 max가 갱신될 뿐 3개값의 max를 구할 수 없는 것이다.

max = Math.max(p1_cnt, p2_cnt);
max = Math.max(max, p3_cnt);

이렇게 하면 됐을 것이다. p1_cnt, p2_cnt 비교한 값 max를 가지고 p3_cnt와 비교하기 때문이다.

 

3. 코드

import java.io.*;
import java.util.*;

class Solution {
    
    static int p1_cnt, p2_cnt, p3_cnt, max;
    static List<Integer> ansList = new ArrayList<> ();
    
    static int[] ans;
    static int[] p1 = {1, 2, 3, 4, 5};
    static int[] p2 = {2, 1, 2, 3, 2, 4, 2, 5};
    static int[] p3 = {3, 3, 1, 1, 2, 2, 4, 4, 5, 5};
    
    public int[] solution(int[] answers) {
        for (int i=0; i<answers.length; i++) {
            //1번
            if (answers[i] == p1[i%5]) p1_cnt++;
            //2번
            if (answers[i] == p2[i%8]) p2_cnt++;
            //3번
            if (answers[i] == p3[i%10]) p3_cnt++;
            //비교
            max = Math.max(p1_cnt, Math.max(p2_cnt, p3_cnt));
            
            // 이렇게 했다가 틀림
            // max = Math.max(p1_cnt, p2_cnt);
            // max = Math.max(p2_cnt, p3_cnt);
            // max = Math.max(p3_cnt, p1_cnt);
        }
        
        // max가 0일 경우
        if (max == 0) return new int[] {1, 2, 3};
        
        if (max == p1_cnt) {
            ansList.add(1);
        } 
        if (max == p2_cnt) {
            ansList.add(2);
        }
        if (max == p3_cnt) {
            ansList.add(3);
        }
        
        ans = new int[ansList.size()];
        for (int i=0; i<ansList.size(); i++) {
            ans[i] = ansList.get(i);
        }
        
        return ans;
    }
}

4. 사담

놓친 부분을 고민하는 시간에 다른 문제를 풀 수 있었을 텐데

처음에 짤 때부터 빠르게 알아차리지 못한 점이 아쉬웠다.

아자좌 또 다른거 풀러 가보자자자