[백준] 행렬

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

1080번: 행렬

 

(0, 0) 부터 시작하여 (N-3, M-3) 까지 비교하면서 더는 바꿀 수 없는 구간에 대해서 A[][] 와 B[][] 가 다르면 -1 을 반환한다. 예를 들면 0 열부터 M-3 까지 비교하여 다음 행 차례가 오면 비교했던 행에 M-2 열과 M-1 이 바꿀 수 없는 구간이다.

 

N 이나 M 이 3보다 작으면 3*3 크기의 부분 행렬이 없으므로 같으면 0 다르면 -1 을 반환한다. 이 예외 때문에 계속 틀렸었다.

 

import sys

N, M = list(map(int, sys.stdin.readline().split()))
A = []
B = []
answer = 0

def change(A, i, j):
    for a in range(3):
        for b in range(3):
            value = not A[i + a][j + b]
            A[i+a][j+b] = value

for i in range(N):
    A.append(list(map(lambda x : x == '1', list(sys.stdin.readline().strip()))))

for i in range(N):
    B.append(list(map(lambda x : x == '1', list(sys.stdin.readline().strip()))))

if N >= 3 and M >= 3:

    for i in range(N-2):
        for j in range(M-2):
            if A[i][j] != B[i][j]:
                change(A, i, j)
                answer += 1

            if i == N - 3:
                if not (A[i+1][j] == B[i+1][j] and A[i+2][j] == B[i+2][j]):
                    answer = -1
                    break

            if j == M - 3:
                if not (A[i][j+1] == B[i][j+1] and A[i][j+2] == B[i][j+2]):
                    answer = -1
                    break

            if i == N - 3 and j == M - 3:
                if not (A[N-1][M-1] == B[N-1][M-1] and A[N-2][M-1] == B[N-2][M-1] and A[N-1][M-2] == B[N-1][M-2] and A[N-2][M-2] == B[N-2][M-2]):
                    answer = -1
                    break

        if answer == -1: break
else:
    if A != B:
        answer = -1

print(answer)