BOJ 1051번 - 숫자 정사각형

[백준 온라인 저지 링크]

1051번: 숫자 정사각형

(0, 0)부터 시작해 2차원 배열의 각 인덱스를 순회하며 가능한 정사각형을 구한 뒤, 꼭지점의 값들이 모두 같은지 여부를 체크하면 됩니다.

구현 아이디어가 어려운 편은 아니지만, 정사각형의 인덱스가 원래 배열을 넘어서지 않게 하는데 애를 꽤나 먹었었네요.

(결국 이 부분은 조건문을 별도로 두어 처리했습니다.)

 

또, 명시되지 않은 부분이지만 입력으로 1 * 1 배열이 주어지면 0이 아닌 1을 출력해야 합니다.

 

[정답 코드 - Python]

N, M = map(int, input().split())
square = []
max_size = 1

for _ in range(N):
    square.append(list(map(int, input())))


def findSquare(row, col):
    size = 0
    for i in range(1, min(N, M) - min(row, col)):
        if row + i >= N or col + i >= M: continue
        if square[row][col] == square[row + i][col] == square[row][
                col + i] == square[row + i][col + i]:
            size = i + 1
    return size * size


for row in range(N - 1):
    for col in range(M - 1):
        size = findSquare(row, col)
        if max_size <= size:
            max_size = size

print(max_size)

+ 여담이지만 제 기준으로는 깔끔하게 구현한 것 같아 뿌듯했던 문제였습니다.