[BOJ] 1051 | 숫자 정사각형

1051 | 숫자 정사각형
https://www.acmicpc.net/problem/1051

 

 

 

 

작성 코드

 

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

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

for k in range(1, min(n, m)+1) :
    for x in range(n-k) :
        for y in range(m-k) :
            if (nums[x][y] == nums[x][y+k] == nums[x+k][y] == nums[x+k][y+k]):
                same.append(k+1)
if same:
    print(max(same)**2)

else:
    print(1)

 

• x, y 는 가로, 세로 인덱스 / k 는 정사각형의 길이
• same 배열에 원소가 들어있으면 max 값을 제곱해서 print, 없다면 1(한 변의 길이-숫자 한 개) print