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

// patterns라는 배열에 수포자들이 찍는 패턴을 intArray 형식으로 저장
val patterns = arrayOf(
    intArrayOf(1, 2, 3, 4, 5), 
    intArrayOf(2, 1, 2, 3, 2, 4, 2, 5), 
    intArrayOf(3, 3, 1, 1, 2, 2, 4, 4, 5, 5)
)

// .foldIndexed()를 사용해서 초기값으로 3칸짜리 IntArray를 가지고 answers를 순회
// 람다식의 파라미터로 인덱스와 누적값, 요소값을 사용
fun solution(answers: IntArray) = answers.foldIndexed(IntArray(3)) { idx, sum, answer -> sum.apply {

// 루프 내부에서 patterns를 순회하며 정답과 일치하는지 확인하여 정답 갯수를 계산합니다.
patterns.forEachIndexed { i, pattern -> if (pattern[idx % pattern.size] == answer) sum[i]++ }

 

// foldIndexed가 종료되면 맞춘 답 갯수가 반환되고, 이를 가지고 최대값만 걸러내 제출
}}.let { list -> (1..3).filter { i -> list[i - 1] == list.maxOf { it } } }

 

전체 코드

class Solution {
    val patterns = arrayOf(
    	intArrayOf(1, 2, 3, 4, 5), 
        intArrayOf(2, 1, 2, 3, 2, 4, 2, 5), 
        intArrayOf(3, 3, 1, 1, 2, 2, 4, 4, 5, 5)
    )
    
    fun solution(answers: IntArray) = answers.foldIndexed(IntArray(3)) { idx, sum, answer -> sum.apply {
        patterns.forEachIndexed { i, pattern -> if (pattern[idx % pattern.size] == answer) sum[i]++ }
    }}.let { list -> (1..3).filter { i -> list[i - 1] == list.maxOf { it } } }
}