Python/브루트 포스

99클럽 코테 스터디 12일차 TIL + 백준 1051 숫자 정사각형 (파이썬)

유일리 2025. 2. 4. 12:57

※ 1051 숫자 정사각형

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

문제 해결 TIP

i를 길이 변수로 두고 j, k 좌표에서 떨어진 좌표를 비교해본다.

 

전체 코드

N, M = map(int,input().split())
rec = []
for i in range(N):
    rec.append(list(map(int,input())))

side = 0
for i in range(min(N, M), 0 , -1):
    for j in range(N - i):
        for k in range(M - i):
            if rec[j][k] == rec[j][k+i] == rec[j+i][k] == rec[j+i][k+i]:
                side = max(side,i+1)
if side != 0:
    print((side)**2)
else:
    print(1)