본문 바로가기
백준 알고리즘

파이썬) 백준 알고리즘 | 1051번 : 숫자 정사각형

by 코딩새내기_ 2022. 1. 30.

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

 

1051번: 숫자 정사각형

N×M크기의 직사각형이 있다. 각 칸에는 한 자리 숫자가 적혀 있다. 이 직사각형에서 꼭짓점에 쓰여 있는 수가 모두 같은 가장 큰 정사각형을 찾는 프로그램을 작성하시오. 이때, 정사각형은 행

www.acmicpc.net

문제

N×M크기의 직사각형이 있다. 각 칸에는 한 자리 숫자가 적혀 있다. 이 직사각형에서 꼭짓점에 쓰여 있는 수가 모두 같은 가장 큰 정사각형을 찾는 프로그램을 작성하시오. 이때, 정사각형은 행 또는 열에 평행해야 한다.

입력

첫째 줄에 N과 M이 주어진다. N과 M은 50보다 작거나 같은 자연수이다. 둘째 줄부터 N개의 줄에 수가 주어진다.

출력

첫째 줄에 정답 정사각형의 크기를 출력한다.

백준 1051 '숫자 정사각형'입니다.

저는 for loop를 3번 돌려서 다 따져봤습니다.

import sys

N, M = map(int, sys.stdin.readline().split())
numbers = []
for i in range(N):
    a = sys.stdin.readline().strip()
    line = []
    for j in range(len(a)):
        line.append(int(a[j]))
    numbers.append(line)
size = 1
for k in range(1,min(N,M)):
    for i in range(N-k):
        for j in range(M-k):
            if numbers[i][j] == numbers[i+k][j] == numbers[i][j+k] == numbers[i+k][j+k]:
                size = k+1

print(size**2)

댓글