[프로그래머스_Kotlin] 행렬의 덧셈

프로그래머스

문제

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

제한 조건	입출력 예

 

내 생각

2중 for문을 사용해서 더하면 될거같다,,

 

풀어봤는데 실패했다. var로 생성해준 answer의 arrayOf 수정

var answer = Array<IntArray>(arr1.size) {i -> IntArray(arr1[i].size)}

 

풀이

//내 풀이
class Solution {
    fun solution(arr1: Array<IntArray>, arr2: Array<IntArray>): Array<IntArray> {
        var answer = Array<IntArray>(arr1.size) { i -> IntArray(arr1[i].size) }
        for (i in arr1.indices) {
            for (j in arr1[i].indices) {
                answer[i][j] = arr1[i][j] + arr2[i][j]
            }
        }

        return answer
    }
}​

//다른 사람의 풀이
class Solution {
    fun solution(arr1: Array<IntArray>, arr2: Array<IntArray>): Array<IntArray> {
        return Array(arr1.size) {
            row ->
            IntArray(arr1[0].size) {
                col ->
                arr1[row][col] + arr2[row][col]
            }
        }
    }
}​

잘하는 사람 엄청 많네,,