[백준][파이썬]1051번 숫자 정사각형

문제 출처 : www.acmicpc.net/problem/1051

1051번: 숫자 정사각형

문제

N*M크기의 직사각형이 있다. 각 칸은 한 자리 숫자가 적혀 있다. 이 직사각형에서 꼭짓점에 쓰여 있는 수가 모두 같은 가장 큰 정사각형을 찾는 프로그램을 작성하시오. 이때, 정사각형은 행 또는 열에 평행해야 한다.

입력

첫째 줄에 N과 M이 주어진다. N과 M은 50보다 작거나 같은 자연수이다. 둘째 줄부터 N개의 줄에 수가 주어진다.

출력

첫째 줄에 정답 정사각형의 크기를 출력한다.

 

내 풀이

n, m = map(int, input().split())

arr = []

for i in range(n):
    arr.append(list(input()))

check = min(n, m)
answer = 0
for i in range(n):
    for j in range(m):
        for k in range(check):
            if ((i + k) < n) and ((j + k) < m) and (arr[i][j] == arr[i][j + k] == arr[i + k][j] == arr[i + k][j + k]):
                answer = max(answer, (k + 1)**2)
                
print(answer)

모든 좌표를 돌면서 k개씩 늘려가며 각 꼭지점의 수가 같은지 판단하고 같을 경우

answer와 비교해서 더 큰 수를 저장하는 방식으로 구현했다.