[프로그래머스] 완전탐색. 수포자

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

문제 설명
수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 1번 문제부터 마지막 문제까지 다음과 같이 찍습니다.
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, ...
1번 문제부터 마지막 문제까지의 정답이 순서대로 들은 배열 answers가 주어졌을 때, 가장 많은 문제를 맞힌 사람이 누구인지 배열에 담아 return 하도록 solution 함수를 작성해주세요.

제한 조건
시험은 최대 10,000 문제로 구성되어있습니다.문제의 정답은 1, 2, 3, 4, 5중 하나입니다.가장 높은 점수를 받은 사람이 여럿일 경우, return하는 값을 오름차순 정렬해주세요.

입출력 예

answers	return
[1,2,3,4,5]	[1]
[1,3,2,4,2]	[1,2,3]

입출력 예 설명

입출력 예 #1
수포자 1은 모든 문제를 맞혔습니다.
수포자 2는 모든 문제를 틀렸습니다.
수포자 3은 모든 문제를 틀렸습니다.
따라서 가장 문제를 많이 맞힌 사람은 수포자 1입니다.

입출력 예 #2
모든 사람이 2문제씩을 맞췄습니다.

 

이번에는 간단한 완전탐색 문제입니다. 다들 이 문제는 어떻게 푸실지 감이 오실거에요.

각 학생별로 패턴을 배열로 선언하고, for loop를 돌면서 문제가 맞을 경우 그 학생의 점수를 관리해주는 변수를 증가 시키면 되겠습니다.

 

처음에는 문제의 정답을 리턴하기 위해 많은 if 문을 사용 하였는데요. 가독성이 너무 좋지 않아 List를 선언하고 최대 점수를 가진 학생일 경우 add를 해주도록 하였습니다.

 

하지만 이렇게 되면 List를 다시 array로 변환 시켜야 하기 때문에 0.x ms 시간이 최대 3.x ms 까지 걸리게 됩니다.

 

알고리즘 문제는 시간이 문제 풀이에 중요하기 때문에 이런 부분들은 체크를 해보면 좋을거 같아요.

저는 채점 후 결과를 보고 가독성을 위해 List로 제출하였습니다.

 

다음은 소스 코드 입니다.

 

public class Supoja {
    public int[] solution(int[] answers) {
        final int[] student1 = {1, 2, 3, 4, 5};
        final int[] student2 = {2, 1, 2, 3, 2, 4, 2, 5};
        final int[] student3 = {3, 3, 1, 1, 2, 2, 4, 4, 5, 5};

        int studentAnswer1 = 0;
        int studentAnswer2 = 0;
        int studentAnswer3 = 0;

        List<Integer> result = new ArrayList<>();

        for (int i = 0; i < answers.length; i++) {
            if (answers[i] == student1[i % student1.length]) {
                studentAnswer1++;
            }
            if (answers[i] == student2[i % student2.length]) {
                studentAnswer2++;
            }
            if (answers[i] == student3[i % student3.length]) {
                studentAnswer3++;
            }
        }

        int topScore = Math.max(studentAnswer1, Math.max(studentAnswer2, studentAnswer3));

        if (topScore == studentAnswer1) {
            result.add(1);
        }
        if (topScore == studentAnswer2) {
            result.add(2);
        }
        if (topScore == studentAnswer3) {
            result.add(3);
        }

        return result.stream().mapToInt(i -> i).toArray();
    }
}

학생 1,2,3의 점수를 모두 분기해주는 if문을 만들면 0.xx ms 시간이 걸립니다.