[프로그래머스] (LV2) 카펫

JavaScript, Python3

문제 설명

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

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

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

제한사항

• 갈색 격자의 수 brown은 8 이상 5,000 이하인 자연수입니다.
• 노란색 격자의 수 yellow는 1 이상 2,000,000 이하인 자연수입니다.
• 카펫의 가로 길이는 세로 길이와 같거나, 세로 길이보다 깁니다.

입출력 예

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

출처

※ 공지 - 2020년 2월 3일 테스트케이스가 추가되었습니다.\
※ 공지 - 2020년 5월 11일 웹접근성을 고려하여 빨간색을 노란색으로 수정하였습니다.

풀이

---

javascript

function solution(brown, yellow) {
    
    for(let y_width = yellow; y_width >= yellow/y_width; y_width--){
        const y_height = yellow/y_width
        if(!Number.isInteger(y_height)){
            continue;
        }
        
        if(brown === ((y_width + y_height)*2 + 4))
            return [y_width+2,y_height+2]
    }
}

python

def solution(brown, yellow):
    for i in range (1, int(yellow ** (1/2)) + 1):
        if yellow % i == 0 and brown == (i*2 + yellow/i*2 + 4):
            return [(yellow/i)+2 , i+2]

설명

---

풀이 전략

완전탐색의 원리대로 모든 경우의 수를 찾으며 푸는 문제.

풀이 설명

[Python 풀이 기준]

• 노란블럭 개수 = y

• 갈색블럭 개수 = b

• 노란블럭의 가로의 길이를 i로 가정하고 1부터 y의 제곱근(정사각형일 경우)까지 반복문을 시행한다.

  • 이때 y & i != 0인 경우는 포함될 수 없으므로 제외한다.
• b의 개수가 i(가로) * 2 + y/i(세로) * 2 + 4(모서리)가 된다면 [y/i+2,i+2]를 반환하고 종료한다.