[코테] 백준 수학 9613번 문제

9613번 : GCD합

문제 링크

https://www.acmicpc.net/problem/9613

문제 설명

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

입력

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

출력

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

문제풀이

itertools의 combinations, math 라이브러리를 활용하여 쉽게 풀이할 수 있다.

from itertools import combinations
import math
import sys
input = sys.stdin.readline

T = int(input())
for _ in range(T) :
    N, *arr = list(map(int, input().split()))
    result = 0 
    for comb in combinations(arr, 2) :
        result += math.gcd(comb[0], comb[1])
    print(result)