본문 바로가기

Mobile/Kotlin

[Kotlin-문법] 행렬 덧셈하는 방법

반응형

fun solution(arrFirst: Array<IntArray>, arrSecond: Array<IntArray>): Array<IntArray> {
	return Array(arrFirst.size) { row ->
    	IntArray(arrFirst[0].size) { col ->
        	arrFrist[row][col] + arrSecond[row][col]
        }
    }
}


fun main() {
    val arrFirst: Array<IntArray> = arrayOf(intArrayOf(1, 2), intArrayOf(2, 3))
    val arrSecond: Array<IntArray> = arrayOf(intArrayOf(3, 4), intArrayOf(5, 6))
    println(solution(arrFirst, arrSecond).contentDeepToString())
}

// 결과값: [[4, 6], [7, 9]]
반응형