[python] 백준 - 1051. 숫자 정사각형

문제 해결

1. (부르트포스) 완전탐색 문제이다.

2. '가장 큰' 정사각형을 구하라고 했으므로 만들수 있는 최대 크기에서부터 하나씩 줄여나갔다.(시간 효율)

3. 네 꼭지점의 크기가 같을 때 return + break를 걸어주어 바로 탐색을 종료

 

+ 난이도가 적당한 2차원배열 + 완전탐색 문제이다.

소스 코드

def find_squre(s):
    # 정사각형의 꼭지점의 숫자 크기가 같은 경우를 찾는다.
    for i in range(n-s+1):
        for j in range(m-s+1):
            if numbers[i][j] == numbers[i][j+s-1] == numbers[i+s-1][j] == numbers[i+s-1][j+s-1]:
                return True

    return False


n, m = map(int, input().split())
numbers = [list(map(int, list(input()))) for _ in range(n)]

# size: 정사각형을 만들 수 있는 최대 크기
# size = m if n > m else size = n
if n > m:
    size = m
else:
    size = n

# 최대 크기부터 하나씩 줄여가며 시작
for k in range(size, 0, -1):
    # 네 꼭지점의 크기가 같은 정사각형을 찾았으면 True를 받아 넓이를 출력해주고 break
    if find_squre(k):
        print(k**2)
        break

 

출처: BACKJOON ONLINE JUDGE

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

1051번: 숫자 정사각형

문제 N*M크기의 직사각형이 있다. 각 칸은 한 자리 숫자가 적혀 있다. 이 직사각형에서 꼭짓점에 쓰여 있는 수가 모두 같은 가장 큰 정사각형을 찾는 프로그램을 작성하시오. 이때, 정사각형은 행 또는 열에 평행해야 한다. 입력 첫째 줄에 N과 M이 주어진다. N과 M은 50보다 작거나 같은 자연수이다. 둘째 줄부터 N개의 줄에 수가 주어진다. 출력 첫째 줄에 정답 정사각형의 크기를 출력한다.