백준 1080번: 행렬 (파이썬)

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

1080번: 행렬

 

문제

0과 1로만 이루어진 행렬 A와 행렬 B가 있다. 이때, 행렬 A를 행렬 B로 바꾸는데 필요한 연산의 횟수의 최솟값을 구하는 프로그램을 작성하시오.

행렬을 변환하는 연산은 어떤 3×3크기의 부분 행렬에 있는 모든 원소를 뒤집는 것이다. (0 → 1, 1 → 0)

입력

 

첫째 줄에 행렬의 크기 N M이 주어진다. N과 M은 50보다 작거나 같은 자연수이다. 둘째 줄부터 N개의 줄에는 행렬 A가 주어지고, 그 다음줄부터 N개의 줄에는 행렬 B가 주어진다.

출력

 

첫째 줄에 문제의 정답을 출력한다. 만약 A를 B로 바꿀 수 없다면 -1을 출력한다.

 

풀이

 

함께 뒤집어지는 다른 구역은 신경쓰지 않고, 각 위치에서 정확히 한 번씩 A와 B를 비교하면 된다.

좌상단부터 우하단으로 내려오면서 현재 위치의 A와 B가 서로 다르면 현재 위치부터 우하단으로 3X3만큼 뒤집도록 한다.

좌상단부터 우하단으로 내려오며 우하단의 3X3만큼 뒤집기 때문에, 이미 확인한 위치는 다시 뒤집지 않는다.

뒤집기 과정이 끝난 후에 두 행렬을 비교하여 같으면 횟수를 출력하고, 다르면 -1을 출력하도록 한다.

 

소스 코드

 

n, m = input().split()
n, m = int(n), int(m)

a = []
b = []
for i in range(n):
    a.append(list(input()))
for i in range(n):
    b.append(list(input()))

cnt = 0
for i in range(n-2):
    for j in range(m-2):
        if a[i][j] != b[i][j]:
            for k in range(i, i+3):
                for l in range(j, j+3):
                    a[k][l] = '0' if a[k][l] == '1' else '1'
            cnt += 1

same = True
for i in range(n):
    for j in range(m):
        if a[i][j] != b[i][j]:
            same = False
            break

if same:
    print(cnt)
else:
    print(-1)