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

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

프로그래머스

문제

행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다. 2개의 행렬 arr1과 arr2를 입력받아, 행렬 덧셈의 결과를 반환하는 함수, solution을 완성해 주세요.

제한 조건
행렬 arr1, arr2의 행과 열의 길이는 500을 넘지 않습니다.

 

풀이

class Solution {
    fun solution(arr1: Array<IntArray>, arr2: Array<IntArray>): Array<IntArray> {
        return Array(arr1.size) { idx1 ->
            IntArray(arr1[idx1].size) { idx2 ->
                arr1[idx1][idx2] + arr2[idx1][idx2]
            }
        }
    }
}

• 결과적으로 같은 위치에 위치하는 2배열의 합으로 구성돼있는 배열을 반환하라는 것입니다. 그렇다면 배열의 생성자를 이용한다면 더욱 쉽게 문제를 풀 수 있습니다.

결과