[백준 2501] 약수 구하기 - 자바(Java)

문제

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

2501번: 약수 구하기

 

 

코드

첫 번째 줄에 정수 n과 k가 공백(" ")으로 구분되어 주어졌을 때, n의 약수들 중 k번째로 작은 수를 출력하는 문제다.

• k번째 약수가 존재하지 않을 경우 0을 출력한다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        int n = Integer.parseInt(st.nextToken());
        int k = Integer.parseInt(st.nextToken());

        int count = generateDivisorCount(n);

        if (count < k) { // 약수의 개수가 k보다 작으면 0 출력
            System.out.println(0);
        } else { // k 번째 약수 출력
            int[] divisors = generateDivisors(n, count);
            System.out.println(divisors[k - 1]);
        }
    }

    // 약수의 개수 구하기
    private static int generateDivisorCount(int n) {
        int count = 0;

        for (int i = 1; i <= Math.sqrt(n); i++) {
            if (i * i == n) {
                count++;
            } else if (n % i == 0) {
                count += 2;
            }
        }

        return count;
    }

    // 약수를 담은 배열 만들기
    private static int[] generateDivisors(int n, int count) {
        int[] divisors = new int[count];
        int idx = 0;

        for (int i = 1; i <= Math.sqrt(n); i++) {
            if (i * i == n) {
                divisors[idx] = i;
                idx++;
            } else if (n % i == 0) {
                divisors[idx] = i;
                divisors[divisors.length - idx - 1] = n / i;
                idx++;
            }
        }

        return divisors;
    }
}