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

from functools import reduce 를 이용해서 여러 원소들의 gcd. 즉, 모든 원소들의 최대공약수를 구하기 ! 

 

Question

 

철수와 영희는 선생님으로부터 숫자가 하나씩 적힌 카드들을 절반씩 나눠서 가진 후, 다음 두 조건 중 하나를 만족하는 가장 큰 양의 정수 a의 값을 구하려고 합니다.

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

 

철수가 가진 카드에 적힌 숫자들을 나타내는 정수 배열 arrayA와 영희가 가진 카드에 적힌 숫자들을 나타내는 정수 배열 arrayB가 주어졌을 때, 주어진 조건을 만족하는 가장 큰 양의 정수 a를 return하도록 solution 함수를 완성해 주세요. 만약, 조건을 만족하는 a가 없다면, 0을 return 해 주세요.

제한사항

• 1 ≤ arrayA의 길이 = arrayB의 길이 ≤ 500,000
• 1 ≤ arrayA의 원소, arrayB의 원소 ≤ 100,000,000
• arrayA와 arrayB에는 중복된 원소가 있을 수 있습니다.

 

Code (After)

 

from math import gcd
from functools import reduce
def solution(A, B):
    answer = []
    def check(a, other):
        for i in other:
            if i%a == 0: return -1
        return a
    
    answer.append(check(reduce(gcd, A), B))
    answer.append(check(reduce(gcd, B), A))
    
    return max(answer) if max(answer)!=-1 else 0

 

Code (Before)

 

from math import gcd
from functools import reduce
def solution(arrayA, arrayB):
    answer = []
    
    def check(gcd_value, array):
        for num in array:
            if num % gcd_value == 0 : return False
        return True
    
    gcd1 = reduce(gcd, arrayA)
    p1 = check(gcd1, arrayB)
    gcd2 = reduce(gcd, arrayB)
    p2 = check(gcd2, arrayA)
    
    if p1&p2 : return max(gcd1, gcd2)
    elif p1 : return gcd1
    elif p2 : return gcd2
    else : return 0
    
    return 0

 

now me

On my github