[알고리즘/백준] 1051번 : 숫자 정사각형(python)

최소가 1이기 때문에 1을 지정해주고 하나씩 늘려가면서 비교한다.
SET형을 사용하여 비교하면 편하다.
set 안에는 하나의 자료형만 들어가야함

N, M = map(int, input().split())
a = [list(input()) for _ in range(N)]
ans = 1
for i in range(N):
    for j in range(M):
        for k in range(ans, min(N, M)):
            if N <= i + k or M <= j + k:
                break
            if len(set([a[i][j], a[i+k][j], a[i][j+k], a[i+k][j+k]])) == 1:
                ans = max(ans, k+1)
print(ans ** 2)