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

해당 글은 백준 1051번 문제 '숫자 정사각형'을 다룬다.

문제

https://www.acmicpc.net/problem/1051

1051번: 숫자 정사각형

 

설명

3중 for문으로 범위만 잘 설정하면 해결할 수 있는 문제이다.

정사각형 크기는 2~min(n, m)까지의 크기를 가질 수 있고

정사각형의 왼쪽 꼭짓점은 행은 0~n-s, 열은 0~m-s의 크기를 가진다.

 

코드

n, m = map(int, input().split())
square = []
for i in range(n):
    square.append(list(map(int, input())))

result = 1          # a b
                    # c d
for s in range(2, min(n, m)+1):
    for i in range(0, n-s+1):
        for j in range(0, m-s+1):
            a = square[i][j]
            b = square[i][j+s-1]
            c = square[i+s-1][j]
            d = square[i+s-1][j+s-1]
            if a == b and b == c and c == d:
                result = s
print(result*result)