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

문제설명

수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 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 함수를 작성해주세요.

class Solution {
    fun solution(answers: IntArray): IntArray {
        val 수포자1 = intArrayOf(1, 2, 3, 4, 5)
        val 수포자2 = intArrayOf(2, 1, 2, 3, 2, 4, 2, 5)
        val 수포자3 = intArrayOf(3, 3, 1, 1, 2, 2, 4, 4, 5, 5)        
        var 수포자1_정답개수 = 0
        var 수포자2_정답개수 = 0    
        var 수포자3_정답개수 = 0
        
        answers.forEachIndexed { index, item ->
            if (item == 수포자1[index % 수포자1.size]) 수포자1_정답개수++
            if (item == 수포자2[index % 수포자2.size]) 수포자2_정답개수++
            if (item == 수포자3[index % 수포자3.size]) 수포자3_정답개수++                
        }
                                    
        val resultMap = mapOf(
            1 to 수포자1_정답개수,
            2 to 수포자2_정답개수,
            3 to 수포자3_정답개수,
        )

        // 맵에서 최대값 찾기
        val maxValue = resultMap.values.maxOrNull()

        // 최대값과 동일한 값을 가지는 키들 찾기
        val answer = resultMap.filterValues { it == maxValue }.keys
                
        return answer.toIntArray()
    }
}

풀이설명

// 1. 문제의 수 만큼 반복을 진행한다.
// 2. 수포자의 찍는 방식의 배열과 정답 수를 저장할 변수를 만든다.
// ex val 수포자1 = [2,1,2,3,2,4,2,5]
// 3. 문제[N] == 수포자N[N]의 값이 같은지 비교한다. 같을경우 해당 수포자의 정답 갯수를 올려준다.
// 4. 맵을 만들어서 1,2,3의 키와 해당되는 정답 개수를 값에 넣어준다.
// 5. 맵에서 최대 값을 찾는다.
// 6. 맵에서 값을 기준으로 5번에서 뽑은 최대값과 동일한 값을 가진 키 리스트를 가져온다.
// 4. 최대값을 answer에 넣고 정렬한다.(여럿일 수 있으니까.)