bright_code
2021. 4. 10. 11:15
728x90
반응형
programmers.co.kr/learn/courses/30/lessons/42842
코딩테스트 연습 - 카펫
Leo는 카펫을 사러 갔다가 아래 그림과 같이 중앙에는 노란색으로 칠해져 있고 테두리 1줄은 갈색으로 칠해져 있는 격자 모양 카펫을 봤습니다. Leo는 집으로 돌아와서 아까 본 카펫의 노란색과
programmers.co.kr
def solution(brown, yellow):
answer = []
total = brown + yellow
p = set()
for i in range(1,total+1):
if total % i == 0 :
p.add( (i, total//i))
for i,j in p :
if i >= j :
if (i-2)*(j-2) ==yellow:
answer.append(i)
answer.append(j)
break
return answer
728x90
반응형