[python/파이썬] 백준 9613 GCD 합

문제

양의 정수 n개가 주어졌을 때, 가능한 모든 쌍의 GCD의 합을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 테스트 케이스의 개수 t (1 ≤ t ≤ 100)이 주어진다. 각 테스트 케이스는 한 줄로 이루어져 있다. 각 테스트 케이스는 수의 개수 n (1 < n ≤ 100)가 주어지고, 다음에는 n개의 수가 주어진다. 입력으로 주어지는 수는 1,000,000을 넘지 않는다.

출력

각 테스트 케이스마다 가능한 모든 쌍의 GCD의 합을 출력한다.

예제 입력 1 복사

3
4 10 20 30 40
3 7 5 12
3 125 15 25

예제 출력 1 복사

70
3
35

 

 

[소스 코드]

 

import sys
import math #gcd 사용위해
import itertools #combinations 사용위해

input = sys.stdin.readline

t = int(input()) #테스트케이스 수

for i in range(t): 
    result = 0 #합
    
    arr = list(map(int, input().split()))
    arr = arr[1:] #첫번째 데이터는 개수이므로 없앤다
    
    for a,b in itertools.combinations(arr,2): #arr의 원소들의 조합
        result += math.gcd(a,b) #각 조합들의 gcd값을 result에 더함
    print(result)