[프로그래머스] 행렬의 덧셈 (Kotlin)

728x90

문제


이해

  1. 행렬의 덧셈 ( 행과 열이 같다 )

접근

  1. 두개의 mapIndexed 함수를 활용하여 각각 더해준다음 반환시킨다

풀이

class Solution {
    fun solution(arr1: Array<IntArray>, arr2: Array<IntArray>): Array<IntArray> {
        var answer = arrayOf<IntArray>()

        answer = arr1.mapIndexed { indexArr, ints ->
            ints.mapIndexed { indexInts, i ->
                i + arr2[indexArr][indexInts]
            }.toIntArray()
        }.toTypedArray()

        return answer
    }
}

 

728x90

댓글

Designed by JB FACTORY