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

이미지에 필터를 돌린다는 느낌으로 범위를 설정하고 각 모서리의 값들이 같은지 확인하면 된다.

# ok는 break 용도

N, M = map(int, input().split())
m = min(N, M)
li = []
for _ in range(N):
    li.append(list(map(int, list(input()))))
ok = False

for l in range(m, 0, -1):
    if ok:
        break
    for i in range(N+1-l):
        if ok:
            break
        for j in range(M+1-l):
            if li[i][j] == li[i+l-1][j] == li[i+l-1][j+l-1] == li[i][j+l-1]:
                ans = l
                ok = True
                break
        
print(ans**2)