2738번: 행렬 덧셈

문제 출처 : https://www.acmicpc.net/problem/2738

2738번: 행렬 덧셈

언어 : Kotlin

 

문제 설명 :

N*M크기의 두 행렬 A와 B가 주어졌을 때, 두 행렬을 더하는 프로그램을 작성하시오.

• 첫째 줄에 행렬의 크기 N 과 M이 주어진다. 둘째 줄부터 N개의 줄에 행렬 A의 원소 M개가 차례대로 주어진다. 이어서 N개의 줄에 행렬 B의 원소 M개가 차례대로 주어진다. N과 M은 100보다 작거나 같고, 행렬의 원소는 절댓값이 100보다 작거나 같은 정수이다.
• 첫째 줄부터 N개의 줄에 행렬 A와 B를 더한 행렬을 출력한다. 행렬의 각 원소는 공백으로 구분한다.

 

제한 사항 :

• 시간 제한 : 1초
• 메모리 제한 : 128MB

 

입출력 예 :

입력	출력
3 3 1 1 1 2 2 2 0 1 0 3 3 3 4 4 4 5 5 100	4 4 4 6 6 6 5 6 100

풀이 : 

import java.io.*

fun main(args: Array<String>) {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val bw = BufferedWriter(OutputStreamWriter(System.out))

    val (n, m) = br.readLine().split(" ").map { it.toInt() }
    val a = Array(n) { IntArray(m) }
    val b = Array(n) { IntArray(m) }

    for (i in 0 until n) {
        val answer = br.readLine().split(" ").map { it.toInt() }

        for (j in 0 until m) {
            a[i][j] = answer[j]
        }
    }

    for (i in 0 until n) {
        val answer = br.readLine().split(" ").map { it.toInt() }

        for (j in 0 until m) {
            b[i][j] = answer[j]
        }
    }

    val ab = Array(a.size) { i ->
        bw.write("${IntArray(a[0].size) { j -> a[i][j] + b[i][j] }.joinToString(" ")}\n")
    }

    bw.flush()
    bw.close()
}