99클럽 코테 스터디 4일차 TIL + 프로그래머스 카펫

문제 설명

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

 

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

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

 

제한사항

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

문제 접근 방식

1. 노란색 카펫의 갯수 배열찾기

• 노란색  카펫의 가로 * 세로 값을 찾을 때 약수를 구하는 방법을 떠올렸다.
• 나눠서 나머지가 0이 되는 값이 노란 카펫 넓이에 대한 약수!

2. (가로 + 2) * (세로 + 2) 이 값이 노란색 + 갈색과 동일해 지는 경우 구하기

• 노란 카펫의 약수를 찾았다면 갈색 카펫은 딱 한겹 감싸는 것이므로 전체 카펫의 넓이 = (노란 카펫의 가로+2) * (노란 카펫의 세로+2)

 

코드로 구현하기

def solution(brown, yellow):
    # 1. 노란 카펫의 배열 구하기
    divided = []
    answer = []
    size = brown + yellow
    
    for i in range(1, yellow+1):
        if yellow%i == 0:
            divided.append(i)
    
    # 2. 갈색 + 노란색 = (n+2) * (m+2) 이면 찾았다!
    for i in divided:
        n = yellow/i
        m = i
        candidate_size = (n+2) * (m+2)
        if candidate_size == size:
            answer.append(n+2)
            answer.append(m+2)
            sorted(answer, reverse = True)
            break
            
    return answer

 

 

결과는...

 

성공이다!!!