[백준] 1080번: 행렬 - Kotlin[코틀린]

문제

 

1080번: 행렬

 

---

풀이

 

 행렬 A와 B를 반복문을 통해 비교한다. 만약 값이 다르다면 A 행렬을 해당 위치를 기준으로 3 x 3 부분 행렬을 뒤집고 횟수를 세어준다. 모든 연산을 마친 후 두 행렬이 다르다면 -1을 출력하고 같다면 뒤집은 횟수를 출력한다.

 

---

코드

 

fun main() {
    val br = System.`in`.bufferedReader()
    val bw = System.out.bufferedWriter()

    val (rows, cols) = br.readLine().split(' ').map { it.toInt() }
    val matrix = Array(rows) { br.readLine().map { it - '0' }.toIntArray() }
    val goal = Array(rows) { br.readLine().map { it - '0' }.toIntArray() }

    var cnt = 0
    for(i in 0 until rows - 2) {
        for(j in 0 until cols - 2) {
            if(matrix[i][j] != goal[i][j]) {
                cnt++
                for(r in i until i + 3) {
                    for(c in j until j + 3) {
                        matrix[r][c] = if(matrix[r][c] == 1) 0 else 1
                    }
                }
            }
        }
    }

    for(i in 0 until rows) {
        for(j in 0 until cols) {
            if(matrix[i][j] != goal[i][j]) cnt = -1
        }
        if(cnt == -1) break
    }

    bw.write("$cnt")
    bw.flush()
    bw.close()
    br.close()
}