프로그래머스 - 모의고사 (with Kotlin)

문제 설명
수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 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문제씩을 맞췄습니다.

문제 생각
1,2,3번 수포자들의 찍기 방법의 규칙에 따라 각각 배열을 생성한 후 들어온 정답문항 배열의 크기가 수포자들의 규칙의 배열보다 크다면 다시 규칙을 지켜 증가시키도록 하였습니다.

그 후 정답배열과 수포자들의 배열을 비교하여 각각의 수포자들의 정답 수를 체크하여 그 중 가장 큰 값을 찾은 후 그 큰 값을 통해 다시 가장 많이 맞춘 수포자들을 찾는 방식을 사용하였습니다.

답변 코드

class Solution {
    data class Pair2(var first:Int, var second: Int)
    
    fun solution(answers: IntArray): ArrayList<Int> {
        //순서대로 수포자 1,2,3
        var fs = intArrayOf(1,2,3,4,5)
        var ss = intArrayOf(2,1,2,3,2,4,2,5)
        var ts = intArrayOf(3,3,1,1,2,2,4,4,5,5)
        //가장 많이 맞춘 수포자 배열 (정답부분)
        var ai = arrayListOf<Int>()
        //가장 많이 맞춘 수포자를 찾을 배열
        var tttt = arrayListOf<Pair2>()
        //수포자1,수포자2,수포자3
        tttt.add(Pair2(1,0))
        tttt.add(Pair2(2,0))
        tttt.add(Pair2(3,0))
        
        //정답 배열이 위의 규칙배열보다 길 경우 규칙에 따라 배열 크기 증가 겸 정답이 맞는 지 체크
        for (i in answers.indices) {
            if (answers.size > fs.size) {
                while (fs.size < answers.size) {
                    fs += fs
                }
            } else if (answers.size > ss.size) {
                while (ss.size < answers.size) {
                    ss += ss
                }
            } else if (answers.size > ts.size) {
                while (ts.size < answers.size) {
                    ts += ts
                }
            }


            if (fs[i] == answers[i]) {
                tttt[0].second += 1
            }
            if (ss[i] == answers[i]) {
                tttt[1].second += 1
            }
            if (ts[i] == answers[i]) {
                tttt[2].second += 1
            }
        }
        //가장 큰 값이 뭔지 찾는 변수
        val m = tttt.maxOf { it.second }
        //위 변수를 통해 찾아낸 수포자들과 그 점수의 배열
        val at = tttt.filter { it.second == m }
        
        //위 변수를 통해 진짜 수포자를 찾음
        for (j in at.indices) {
            ai.add(at[j].first)
        }
        
        return ai
    }
}

데이터 클래스로 Pair2를 만들어 기존 Pair의 copy로 값을 변경하는 것 보다 유연하게 사용할 수 있도록 하였습니다.