[Baekjoon] 1051번 숫자 정사각형

문제

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

입력

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

출력

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

예제

Example 1:

Input: 
3 5
42101
22100
22101
Output: 
9

조건

시간 제한 : 2초

메모리 제한 : 128 MB

풀이과정

내 풀이

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

# 정사각형이 될 수 있는 최대치 계산
max_n = min(m,n)

graph = []
for _ in range(n):
    graph.append(list(map(int,input())))

ans = 0

# max_n을 하나씩 줄여가면서 비교, 답이 나온다면 while문 빠져나오기
while(1):
    for i in range(m-max_n+1):
        for j in  range(n-max_n+1):
            if graph[j][i] == graph[j+max_n-1][i] == graph[j][i+max_n-1] == graph[j+max_n-1][i+max_n-1]:
                ans = max_n**2

    if ans != 0:
        break
    max_n -= 1

print(ans)