코드카타 - 행렬의 덧셈

문제 설명
행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다. 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> {
        var answer = arr1
        var len1 = arr1.size
        var len2 = arr1[0].size
        // println(len)
        for (i in 0..len1-1){
            for (j in 0..len2-1){
                answer[i][j] = arr1[i][j] + arr2[i][j]
            }
        }
        return answer
    }
}

 

주의사항

- 행렬의 연산을 할 때에는 행렬 자체의 길이 / 행렬 내 원소의 길이를 따로따로 보고 계산해야 한다. 

  그래서 len1, len2를 따로 놓고 계산함.