백준 알고리즘 1080번: 행렬(Python)

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

1080번: 행렬

---

 

문제 해결을 위한 과정

이 문제의 경우 규칙을 찾으려 노력하다가 규칙이 도저히 발견이 되지 않아서 단순하게 3*3 행렬을 뒤집는 함수를 만든 후 문제를 풀 수 있었습니다. N, M의 범위가 50 이하인 자연수이므로 O(N^2)로 조회하며 모든 좌표에 대해 3*3 행렬을 뒤집는다면 (다만 마지막에서 두 번째 세 번째 열까지 가능합니다.) 50 * 47 * 3* 3 이므로 대략 21150번의 연산을 수행해야 합니다. 파이썬의 경우 1초에 약 2천만 번의 연산을 할 수 있다고 알려져 있으므로 제한시간이 2초인 이 문제에서는 충분히 해결할 수 있는 방법입니다. 따라서 모든 경우의 수를 따져가며 브루트 포스의 방식으로 해결합니다.

---

소스코드

def sumGraph(x, y):
  result = 0
  for i in range(x):
    for j in range(y):
      result += graph3[i][j]
  return result
 
def switch(x, y):
  for i in range(x, x+3):
    for j in range(y, y+3):
      if graph[i][j] == 0:
        graph[i][j] = 1
      else:
        graph[i][j] = 0
 
n, m = map(int, input().split())
graph = []
graph2 = []
graph3 = [[0] * m for _ in range(n)]
count = 0
 
for i in range(n):
  graph.append(list(map(int, input())))
  
for i in range(n):
  graph2.append(list(map(int, input())))
 
if n < 3 and m < 3:
  for i in range(n):
    for j in range(m):
      if graph[i][j] == graph2[i][j]:
        graph3[i][j] = 0
      else:
        graph3[i][j] = 1
  if(sumGraph(n,m))==0:
    print(0)
  else:
    print(-1)
else:
  for i in range(n-2):
    for j in range(m-2):
      if graph[i][j] != graph2[i][j]:
        switch(i, j)
        count += 1
  for i in range(n):
    for j in range(m):
      if graph[i][j] == graph2[i][j]:
        graph3[i][j] = 0
      else:
        graph3[i][j] = 1
  if sumGraph(n, m) != 0:
    print(-1)
  else:
    print(count)