[Programmers] Lv2 - 숫자 카드 나누기

문제

https://school.programmers.co.kr/learn/courses/30/lessons/135807

프로그래머스

 

풀이

1. 철수가 가진 카드들에 적힌 모든 숫자를 나눌 수 있고 영희가 가진 카드들에 적힌 모든 숫자들 중 하나도 나눌 수 없는 양의 정수 a
2. 영희가 가진 카드들에 적힌 모든 숫자를 나눌 수 있고, 철수가 가진 카드들에 적힌 모든 숫자들 중 하나도 나눌 수 없는 양의 정수 a

 

위 조건을 만족하는 수 중 가장 큰 양의 정수를 반환해야한다.

def solution(arrayA, arrayB):
    def gcd(num1, num2):
        if not num2:
            return num1
        return gcd(num2, num1 % num2)

    def take_gcd(array):
        ret = array[0]
        for i in range(1, len(array)):
            ret = gcd(ret, array[i])
        return ret

    def check_condition(gcd_ret, array):
        for num in array:
            if not num % gcd_ret:
                return False
        return True

    a, b = take_gcd(arrayA), take_gcd(arrayB)

    checked_a, checked_b = check_condition(a, arrayB), check_condition(b, arrayA)

    return max((checked_a * a), (checked_b * b))

• gcd: 두 수의 최대공약수를 구하는 함수
• take_gcd: 리스트의 각 요소들을 gcd함수에 넣어 리스트 전체의 최대공약수를 구하는 함수
• check_condition: 위의 조건을 만족하는지 판단하는 함수

철수군와 영희양이 카드를 꼭 두개씩만 가지는 건 아니기때문에, take_gcd를 따로 만들어줬다.

 

먼저 리스트 A, B의 최대공약수를 take_gcd를 통해 구한뒤 a, b에 할당한다.
그 다음 check_condition으로 조건에 부합하는지 확인해본다.

 

경우에 따라서 A, B 모두 조건을 만족할 수 있으므로, 마지막에 둘 중 더 큰값을 확인해서 반환해야한다.
check_condition으로 조건에 부합하면 True, 부합하지 않으면 False를 반환하게된다.
즉 True면 1, False면 0을 반환하는셈이므로 max()을 사용해줬다.
이러면 A, B가 1)둘 다 조건을 만족하지 않을때, 2)둘 중 하나만 조건을 통과할때, 3)둘 다 조건을 만족할때 를 한꺼번에 고려할 수 있다.