[백준][코틀린][1080] 행렬

https://www.acmicpc.net/problem/1080

1080번: 행렬

문제 풀이

1. 주어진 입력의 A 행렬을 matrix 에 B 행렬을 target 에 저장한다.

2. 이중 for 을 이용하여 matrix 행렬을 탐색한다.

3. matrix[i][j] != target[i][j] 인 경우 이면서 i + 2 < N 이고 j + 2 < M 인 경우이면 (i, j) 를 기준으로 3X3 영역의 값들을 (0 -> 1, 1 -> 0) 으로 바꾸고 바꾼 횟수를 count 에 저장한다.

4. matrix 와 target 을 비교하여 서로 같지 않으면 -1을 출력하고 같으면 count 값을 출력한다.

코드

import java.io.*

fun main(args: Array<String>) {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val (N, M) = br.readLine().split(" ").map { it.toInt() }
    val matrix = Array(N) { IntArray(M) { 0 } }
    val target = Array(N) { IntArray(M) { 0 } }
    repeat(N) { index ->
        matrix[index] = br.readLine().map { it.digitToInt() }.toIntArray()
    }

    repeat(N) { index ->
        target[index] = br.readLine().map { it.digitToInt() }.toIntArray()
    }

    var result = 0

    for (i in 0 until N) {
        if (i + 2 >= N) continue

        for (j in 0 until M) {
            if ((i + 2) < N && (j + 2) < M && matrix[i][j] != target[i][j]) {
                // matrix 와 target 이 서로 다르므로 부분 행렬 3X3의 원소들을 업데이트 한다. (0 -> 1, 1 -> 0)
                for (k in i..i + 2) {
                    for (l in j..j + 2) {
                        matrix[k][l] = Math.abs(matrix[k][l] - 1)
                    }
                }

                result++
            }
        }
    }

    var matched = true

    loop@ for (i in 0 until N) {
        for (j in 0 until M) {
            if (matrix[i][j] != target[i][j]) {
                matched = false
                break@loop
            }
        }
    }

    if (matched) println(result)
    else println(-1)
}