코딩두의 포트폴리오

[BAEKJOON 2501] 약수 구하기 본문

Algorithm/BAEKJOON

[BAEKJOON 2501] 약수 구하기

코딩두 2024. 12. 20. 13:46

# 문제 조건

약수 구하기

 

시간 제한메모리 제한제출정답맞힌 사람정답 비율
1 초 128 MB 80392 39791 35049 50.124%

문제

어떤 자연수 p와 q가 있을 때, 만일 p를 q로 나누었을 때 나머지가 0이면 q는 p의 약수이다. 

6을 예로 들면

  • 6 ÷ 1 = 6 … 0
  • 6 ÷ 2 = 3 … 0
  • 6 ÷ 3 = 2 … 0
  • 6 ÷ 4 = 1 … 2
  • 6 ÷ 5 = 1 … 1
  • 6 ÷ 6 = 1 … 0

그래서 6의 약수는 1, 2, 3, 6, 총 네 개이다.

두 개의 자연수 N과 K가 주어졌을 때, N의 약수들 중 K번째로 작은 수를 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 N과 K가 빈칸을 사이에 두고 주어진다. N은 1 이상 10,000 이하이다. K는 1 이상 N 이하이다.

출력

첫째 줄에 N의 약수들 중 K번째로 작은 수를 출력한다. 만일 N의 약수의 개수가 K개보다 적어서 K번째 약수가 존재하지 않을 경우에는 0을 출력하시오.

예제 입력 1 복사

6 3

예제 출력 1 복사

3

예제 입력 2 복사

25 4

예제 출력 2 복사

0

예제 입력 3 복사

2735 1

예제 출력 3 복사

1

 

// 필요한 패키지들 임포트
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Collections;

public class p2501 {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] input = br.readLine().split(" ");
        int N = Integer.parseInt(input[0]);
        int K = Integer.parseInt(input[1]);

        ArrayList<Integer> divisors = new ArrayList<>();

        // 약수 구하기 (최적화)
        for (int i = 1; i * i <= N; i++) {
            if (N % i == 0) {
                divisors.add(i); // 작은 약수 추가
                if (i != N / i) {
                    divisors.add(N / i); // 대응되는 큰 약수 추가
                }
            }
        }

        // 약수를 정렬 (필요한 경우)
        Collections.sort(divisors);

        // K번째 약수 출력
        if (K > divisors.size()) {
            System.out.println(0); // 약수의 개수가 K보다 적은 경우
        } else {
            System.out.println(divisors.get(K - 1));
        }
    }
}

'Algorithm > BAEKJOON' 카테고리의 다른 글

[BAEKJOON 9012] 괄호 - 추후수정  (1) 2024.03.27
[BAEKJOON 1463] 1로 만들기  (0) 2024.03.26