-
문제 설명
Leo는 카펫을 사러 갔다가 아래 그림과 같이 중앙에는 노란색으로 칠해져 있고 테두리 1줄은 갈색으로 칠해져 있는 격자 모양 카펫을 봤습니다.
Leo는 집으로 돌아와서 아까 본 카펫의 노란색과 갈색으로 색칠된 격자의 개수는 기억했지만, 전체 카펫의 크기는 기억하지 못했습니다.
Leo가 본 카펫에서 갈색 격자의 수 brown, 노란색 격자의 수 yellow가 매개변수로 주어질 때 카펫의 가로, 세로 크기를 순서대로 배열에 담아 return 하도록 solution 함수를 작성해주세요.
제한사항- 갈색 격자의 수 brown은 8 이상 5,000 이하인 자연수입니다.
- 노란색 격자의 수 yellow는 1 이상 2,000,000 이하인 자연수입니다.
- 카펫의 가로 길이는 세로 길이와 같거나, 세로 길이보다 깁니다.
10 2 [4, 3] 8 1 [3, 3] 24 24 [8, 6] 코드
def solution(brown, yellow): answer = [] divisors = [] divisors_back = [] divide = [] for i in range(1, int(yellow**(1/2)) + 1): if (yellow % i == 0): divisors.append(i) if (i != (yellow // i)): divisors_back.append(yellow//i) divide = divisors + divisors_back[::-1] if len(divide) == 1: answer = [3,3] else: for i in range(len(divide)): if brown == ((yellow // divide[i]) + divide[i])*2 + 4: temp = [divide[i]+2, yellow // divide[i]+2] answer = [max(temp),min(temp)] return answer
'알고리즘 > 프로그래머스' 카테고리의 다른 글
[프로그래머스] 프렌즈4블록 (파이썬) (0) 2022.01.09 [프로그래머스] 큰 수 만들기 (파이썬) (0) 2022.01.09 [프로그래머스] H-Index (파이썬) (0) 2022.01.08 [프로그래머스] 다리를 지나는 트럭 (파이썬) (0) 2022.01.08 [프로그래머스] 위장 (파이썬) (0) 2022.01.08