📚 출처 - 1780 - 종이의 개수
문제 설명
N×N크기의 행렬로 표현되는 종이가 있다. 종이의 각 칸에는 -1, 0, 1 중 하나가 저장되어 있다. 우리는 이 행렬을 다음과 같은 규칙에 따라 적절한 크기로 자르려고 한다.
이와 같이 종이를 잘랐을 때, -1로만 채워진 종이의 개수, 0으로만 채워진 종이의 개수, 1로만 채워진 종이의 개수를 구해내는 프로그램을 작성하시오.
입력
첫째 줄에 N(1 ≤ N ≤ 37, N은 3k 꼴)이 주어진다. 다음 N개의 줄에는 N개의 정수로 행렬이 주어진다.
출력
첫째 줄에 -1로만 채워진 종이의 개수를, 둘째 줄에 0으로만 채워진 종이의 개수를, 셋째 줄에 1로만 채워진 종이의 개수를 출력한다.
입출력 예
예제 입력 | 예제 출력 |
---|---|
9 0 0 0 1 1 1 -1 -1 -1 0 0 0 1 1 1 -1 -1 -1 0 0 0 1 1 1 -1 -1 -1 1 1 1 0 0 0 0 0 0 1 1 1 0 0 0 0 0 0 1 1 1 0 0 0 0 0 0 0 1 -1 0 1 -1 0 1 -1 0 -1 1 0 1 -1 0 1 -1 0 1 -1 1 0 -1 0 1 -1 | 10 12 11 |
Logic 1
먼저 전체를 확인하고 같지 않은게 있을시에 가로, 세로 3등분 해 다시 비교하는 방법입니다.
def check(x, y, n):
global neg, neut, pos
num_check = paper[x][y]
for i in range(x, x+n):
for j in range(y, y+n):
if paper[i][j] != num_check:
for k in range(3):
for l in range(3):
check(x + k * n//3, y + l * n//3, n//3)
return
if num_check == -1:
neg += 1
elif num_check == 0:
neut += 1
else:
pos += 1
N = int(input())
paper = [list(map(int, input().split())) for _ in range(N)]
neg = 0
neut = 0
pos = 0
check(0, 0, N)
print(f'{neg}\n{neut}\n{pos}')