[파이썬] 카펫

출처 : programmers.co.kr/learn/courses/30/lessons/42842

코딩테스트 연습 - 카펫

 

☆ 나의 풀이

 

# 35/100 실패
def solution(brown, yellow):
    answer = []
    area=brown+yellow
    for i in range(2,area//2):
        if area%i==0:
            answer.append([i,area//i])
    return max(answer)

 

# 통과
def solution(brown, yellow):
    answer = []
    area=brown+yellow
    for i in range(2,area//2):
        if area%i==0 :
            if (i-2)*(area//i-2)==yellow:
                answer.append([i,area//i])
    return max(answer) 

 

★ 다른 사람 풀이

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

 

import math
def solution(brown, yellow):
    w = ((brown+4)/2 + math.sqrt(((brown+4)/2)**2-4*(brown+yellow)))/2
    h = ((brown+4)/2 - math.sqrt(((brown+4)/2)**2-4*(brown+yellow)))/2
    return [w,h]