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

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

프로그래머스

 

풀이 접근 방법

1. 1번,2번,3번 수포자에는 반복되는 부분이 있으므로 반복 되는 부분을 제거하고 배열에 저장한다.

2. answers 배열을 돌면서 각 수포자들이 맞은 개수를 계산한다.

3. 3명의 값을 max 연산을 통해서 비교하고, 최종 max값에 해당하는 수포자를 배열에 넣어준다.

3-2. 3명의 값이 모두 동일한 경우, 셋 다 배열에 넣어준다 (오름차순으로)

 

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 [] three = {3, 3, 1, 1, 2, 2, 4, 4, 5, 5};
                
        int one_ans = 0; int two_ans = 0; int three_ans = 0;
        for(int i = 0; i < answers.length; i++) {
            if (answers[i] == one[i % one.length]) one_ans++;
            if (answers[i] == two[i % two.length]) two_ans++;
            if (answers[i] == three[i % three.length]) three_ans++;
        }
        
        int res = Math.max(one_ans, Math.max(two_ans, three_ans));
        List<Integer> list = new ArrayList<>();
        
        if (one_ans == two_ans && two_ans == three_ans) {
            int [] answer = new int[3];
            answer[0] = 1; answer[1] = 2; answer[2] = 3;
            return answer;
        }
        
        if (res == one_ans) list.add(1);
        if (res == two_ans) list.add(2);
        if (res == three_ans) list.add(3);
        
        int[] answer = new int[list.size()];
        for(int i = 0; i < answer.length; i++) answer[i] = list.get(i);

        return answer;
    }
}

 

풀이 회고

 

일단 세개의 값을 비교하는 것을 처음에 떠올리지 못해서 풀이법을 참고했다.

3개의 값을 비교할 때는, Math.max(ans1, Math.max(ans2, ans3))
이런식으로 두 수의 최대값과 나머지 수의 최대값을 비교해주면 쉽게 해결 가능하다!

 

모두 맞춘 개수가 동일한 경우에 대해서 너무 하드코딩을 한거 같다. 

 

문제엔 시간 제한이 나와있지 않은 것 같은데 answer의 길이가 10,000 이므로 O(N) 연산을 하면 1억이 나오기 때문에 1초가 걸린다.