[Programmers/Python] [PCCE 기출문제] 10번 / 공원

📝 문제 설명

지민이는 다양한 크기의 정사각형 모양 돗자리를 가지고 공원에 소풍을 나왔습니다. 공원에는 이미 돗자리를 깔고 여가를 즐기는 사람들이 많아 지민이가 깔 수 있는 가장 큰 돗자리가 어떤 건지 확인하려 합니다. 예를 들어 지민이가 가지고 있는 돗자리의 한 변 길이가 5, 3, 2 세 종류이고, 사람들이 다음과 같이 앉아 있다면 지민이가 깔 수 있는 가장 큰 돗자리는 3x3 크기입니다.
지민이가 가진 돗자리들의 한 변의 길이들이 담긴 정수 리스트 mats, 현재 공원의 자리 배치도를 의미하는 2차원 문자열 리스트 park가 주어질 때 지민이가 깔 수 있는 가장 큰 돗자리의 한 변 길이를 return 하도록 solution 함수를 완성해 주세요. 아무런 돗자리도 깔 수 없는 경우 -1을 return합니다.

 

 

🤔 문제 풀이

테이블의 -1 값을 가진 셀에서 돗자리 크기가 전체 테이블 범위 안에 있는지 확인한다. 이후 해당 범위의 2차원 배열을 순회하여 모두 -1의 값인지 확인한다. 모두 -1일 경우 돗자리 크기의 값을 result 리스트에 추가하고 리스트에서 가장 큰 값을 return 하면 된다.

 

 

🌟 풀이 코드 

def check(row, col, size, park):
    for i in range(row, row + size):
        for j in range(col, col + size):
            if not (park[i][j] == "-1"):
                return False
    return True

def solution(mats, park):
    result = []
    row = len(park)
    col = len(park[0])
    for i in range(row):
        for j in range(col):
            if park[i][j] == "-1":
                for k in mats:
                    if i + k > row or j + k > col:
                        continue
                    if check(i, j, k, park):
                        result.append(k) 
    if not result:
        answer = -1
    else:
        answer = max(result)
    return answer