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

문제 링크

---

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

1051번: 숫자 정사각형

 

풀이 코드

---

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)

코드 해석

---

만약 i+k가 n보다 작고 j+k가 m보다 작으며 arr[i][j] == arr[i][j + k] == arr[i + k][j] == arr[i + k][j + k] 라면

answer에 가장 큰 수를 제곱하여 저장한다.