[BOJ: 1080] - Python / 파이썬 - 행렬

o_jooon_·2024년 1월 31일
0

BOJ

목록 보기
22/49
post-thumbnail

서론

그리디 문제입니다.
배열을 탐색하며 현재 좌표 기준으로 다른 배열과 동일한지 확인 후, 다르면 뒤집기만 하면 됩니다.

난이도

실버 1


문제

조건

시간 제한메모리 제한
2 초128 MB

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

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


입력

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


출력

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


예시

예제 입력 1

3 4
0000
0010
0000
1001
1011
1001

예제 출력 1

2

예제 입력 2

1 1
1
0

예제 출력 2

-1

예제 입력 3

18 3
001
100
100
000
011
010
100
100
010
010
010
110
101
101
000
110
000
110
001
100
011
000
100
010
011
100
101
101
010
001
010
010
111
110
111
001

예제 출력 3

7

풀이

좌표를 한 번씩만 탐색하며 뒤집는 이유는 이미 뒤집은 좌표를 두 번 뒤집으면 처음과 동일하기 때문에 한 번씩만 뒤집습니다.

  1. 왼쪽 위부터 차례대로 탐색하며 현재 위치가 아니라면 현재 좌표 기준 오른쪽 아래 방향 3x3 행렬을 뒤집는다.
  2. 탐색이 끝난 후에도 두 배열이 일치하지 않으면 -1을 출력, 아니면 뒤집은 수를 출력한다.

코드

import sys
input = sys.stdin.readline

def reverse(x, y):							# 현재 좌표(가장 왼쪽 위) 기준 3 x 3 행렬 뒤집기
    for i in range(x, x + 3):
        for j in range(y, y + 3):
            first[i][j] = 1 - first[i][j]	# 0은 1이 되고 1은 0이 됨

n, m = map(int, input().split())
first = [list(map(int, input().rstrip())) for _ in range(n)]
second = [list(map(int, input().rstrip())) for _ in range(n)]
ans = 0	# 뒤집은 수

for i in range(n - 2):					# 3칸을 뒤집어야 하기 때문에 n-2까지
    for j in range(m - 2):
        if first[i][j] != second[i][j]:	# 현재 좌표의 두 배열 값이 다르면 뒤집기
            reverse(i, j)
            ans += 1

print(ans if first == second else -1)	# 탐색 종료 후 같으면 뒤집은 수, 아니면 -1 출력

실행 결과

profile
안녕하세요.

0개의 댓글