[Programmers] 코딩테스트입문 > 분수의덧셈

Q. 첫 번째 분수의 분자와 분모를 뜻하는 numer1, denom1, 두 번째 분수의 분자와 분모를 뜻하는 numer2, denom2가 매개변수로 주어집니다. 두 분수를 더한 값을 기약 분수로 나타냈을 때 분자와 분모를 순서대로 담은 배열을 return 하도록 solution 함수를 완성해보세요. 

 

위 문제를 해결하기 위해서 생각했던게 두 분수의 합을 계산하기 위해 공통분모를 구하고 분자와 분모를 더한후

최대공약수(GCD) 를 사용하여 기약 분수로 변환하는 방식을 생각했다. 

 

처음 코드를 구현했을때

import math

def solution(numer1, denom1, numer2, denom2):
    # 공통 분모 구하기
    common_denom = denom1 * denom2
    
    # 분자 계산 (첫 번째 분수의 분자는 denom2, 두 번째 분수의 분자는 denom1을 곱해서 통분)
    common_numer = numer1 * denom2 + numer2 * denom1
    
    # 최대공약수 구하기
    gcd_value = math.gcd(common_numer, common_denom)
    
    # 기약 분수로 만들기
    return [common_numer // gcd_value, common_denom // gcd_value]

 

위 방식으로 구현했지만 뭔가 더 간략화 가능할거 같은데...??

고민을 하다가 중간 계산을 한 줄로 처리하고, 불필요한 공식을 최소화 시켜보자 라는 생각으로 아래의 코드를 구현해봤다.

하면서 매개변수의 이름도 짧게 줄여보았다. number1 을 n1 로 denom1 을 d1 로 내 맘대로 줄여보았더니 확실히 위에꺼보다 깔끔해보이긴한데 이게 맞나...??? 맞겠지...맞을거야...

 

import math

def solution(n1, d1, n2, d2):
    return [(n1 * d2 + n2 * d1) // math.gcd(n1 * d2 + n2 * d1, d1 * d2), 
            d1 * d2 // math.gcd(n1 * d2 + n2 * d1, d1 * d2)]

 

확실히 오랜만에 코딩테스트 준비할려고하니까 쉽지는 않네..

화이팅!!!!!!! 가보자고!!!!!!!!!