CS/Algorithm 문제

[Programmers] 카펫

[Programmers] 카펫

 

🎈문제

https://programmers.co.kr/learn/courses/30/lessons/42842

💬설명

  • 근의 공식으로 풀 수 있었던 문제
  • 변을 각각 x, y라고 하면 x + y = (brown + 4) / 2, xy = yellow + brown 이 된다.
  • yellow와 brown을 알고 있으니 x, y를 변수로 둬서 근의 공식으로 풀면된다.

👩‍💻코드

def solution(brown, yellow):
    a = (brown + 4) / 2
    b = yellow + brown
    
    x = (a + (a ** 2 - 4 * b) ** (1/2)) / 2
    y = (a - (a ** 2 - 4 * b) ** (1/2)) / 2
    
    return sorted([x, y], reverse=True)
728x90
반응형