[Programmers] Lv1.모의고사(kotlin)

문제

https://programmers.co.kr/learn/courses/30/lessons/42840?language=kotlin

 

예시

https://programmers.co.kr/learn/courses/30/lessons/42840?language=kotlin

 

풀이

1. numarr 배열에 1번, 2번, 3번 수포자의 찍는 방식을 저장합니다.

2. 수포자가 찍는 방식은 일정 패턴으로 반복되기에 answers의 갯수와 numarr[i]의 갯수가  다르더라도 나머지 연산자를 이용하면 비교할 수 있습니다.

3. 최대 높은 점수를 받은 번호를 체크하고 반환합니다.

 

import kotlin.math.*

class Solution {
    fun solution(answers: IntArray): IntArray {
        var answer = intArrayOf(0,0,0)
        
        val numarr = arrayOf(
            arrayOf(1,2,3,4,5),
            arrayOf(2,1,2,3,2,4,2,5),
            arrayOf(3,3,1,1,2,2,4,4,5,5)
        )
        
        for(i in 0..2){
            var score = 0
            
            for(j in 0 until answers.size){
                score += if(numarr[i][j%numarr[i].size] == answers[j]) 1 else 0
            }
            answer[i] = score
        }
        
        return answer.mapIndexed{ index, i -> 
            if(i == answer.maxOf{it}){
                index+1
            }else{
                -1
            }
        }.filter{ it != -1}.toIntArray()
    }
}

 

 

참고

• 함수형 프로그래밍 언어인 kotlin의 특성에 맞게 함수를 많이 익히자 😽
• https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.collections/map-indexed.html

mapIndexed - Kotlin Programming Language

• https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.collections/max-of.html

maxOf - Kotlin Programming Language