프로그래머스 (Lv 2) - 카펫

1) 문제 설명

• Leo는 카펫을 사러 갔다가 아래 그림과 같이 중앙에는 노란색으로 칠해져 있고 테두리 1줄은 갈색으로 칠해져 있는 격자 모양 카펫을 봤습니다.

 

• Leo는 집으로 돌아와서 아까 본 카펫의 노란색과 갈색으로 색칠된 격자의 개수는 기억했지만, 전체 카펫의 크기는 기억하지 못했습니다.

 

• Leo가 본 카펫에서 갈색 격자의 수 brown, 노란색 격자의 수 yellow가 매개변수로 주어질 때 카펫의 가로, 세로 크기를 순서대로 배열에 담아 return 하도록 solution 함수를 작성해주세요.

 

2) 제한 사항

• 갈색 격자의 수 brown은 8 이상 5,000 이하인 자연수입니다.

 

• 노란색 격자의 수 yellow는 1 이상 2,000,000 이하인 자연수입니다.

 

• 카펫의 가로 길이는 세로 길이와 같거나, 세로 길이보다 깁니다.

 

3) 입출력 예제

brown    yellow    return
10    2    [4, 3]
8    1    [3, 3]
24    24    [8, 6]

 

4) 입출력 예제 설명

• 없음

 

5) 풀이

def solution(brown, yellow):
    # 총 블록 수
    allBlock = brown + yellow

    # 초기 가로, 세로 길이
    width = 3
    height = 1

    while True:
        # 총 블록수 / 가로길이 --> 몫, 나머지
        height = allBlock // width
        remainder = allBlock % width

        # 나머지 0이고 가로 길이 >= 세로 길이 이고
        # 중앙 블록 (가로 -2 * 세로 - 2) == 노랑 블록 수 이면 해당 가로,세로 길이 값 리턴
        if remainder == 0 and width >= height and yellow == (width-2) * (height-2):
            return [width, height]

        # 그게 아니면 가로 길이 +1
        width += 1

 

[Reference]

• https://school.programmers.co.kr/learn/courses/30/lessons/42842

프로그래머스