[백준-1051] 숫자 정사각형 / Python

1051번: 숫자 정사각형

import sys

input = sys.stdin.readline

n, m = map(int, input().split())
arr = [list(map(str, input().rstrip())) for _ in range(n)]
temp = min(n, m)
answer = 0
for i in range(n):
    for j in range(m):
        for k in range(temp):
            if i + k < n and j + k < m:
                if (
                    arr[i][j] == arr[i][j + k]
                    and arr[i][j] == arr[i + k][j]
                    and arr[i][j] == arr[i + k][j + k]
                ):
                    answer = max(answer, (k + 1) ** 2)
print(answer)

 

해설

각 원소 별로 k를 늘려주며 정사각형의 꼭짓점을 확인한다. 이때, 꼭짓점은 전체 직사각형의 범위를 벗어나면 안된다. 해당 정사각형의 한 변의 길이는 k+1이 되므로 너비는 (k+1)**2가 된다. 이 넓이 중 최대 넓이를 출력한다.