[Lv. 1] 행렬의 덧셈

문제 출처 : https://school.programmers.co.kr/learn/courses/30/lessons/12950

프로그래머스

난이도 : Level.1

언어 : Kotlin

 

문제 설명 :

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

 

제한 사항 : 

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

입출력 예 :

arr1	arr2	return
[[1, 2], [2, 3]]	[[3, 4], [5, 6]]	[[4, 6], [7, 9]]
[[1], [2]]	[[3], [4]]	[[4], [6]]

풀이 : 

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