[백준] 9613 - GCD 합 (java)

문제

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

입력

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

출력

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

예제 입력

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

예제 출력

70
3
35

코드

import java.io.*;
import java.util.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        int t = Integer.parseInt(br.readLine());
        for (int i = 0; i < t; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            int n = Integer.parseInt(st.nextToken());
            long[] num = new long[n];
            for (int j = 0; j < n; j++) {
                num[j] = Integer.parseInt(st.nextToken());
            }

            bw.write(sumGCD(num) + "\n");
        }

        br.close();
        bw.close();
    }

    public static long sumGCD(long[] arr) {
        long sum = 0;

        for (int i = 0; i < arr.length; i++) {
            for (int j = i + 1; j < arr.length; j++) {
                sum += GCD(arr[i], arr[j]);
            }
        }

        return sum;
    }

    public static long GCD(long a, long b) {
        if (b == 0) return a;
        else return GCD(b, a % b);
    }
}

정리

sumGCD 함수 내에서 2중 for 문을 돌며 sum에 모든 쌍의 GCD를 더해주고, 이를 return 해줬다.

처음에 long이 아닌 int를 써서 틀렸었다. 자료형 선택에 조금 더 주의하자.