[정올] 1402 : 약수 구하기 (*)

문제

어떤 자연수 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을 출력하시오.

입력 예

6 3

출력 예

3

 

코드

import java.util.ArrayList;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);

        int n = sc.nextInt();
        int k = sc.nextInt();
        int num = 0; // 약수 개수

        // 약수 저장할 배열
        ArrayList<Integer> arr = new ArrayList<Integer>();

        // n의 약수 구하기
        for(int i=1; i<=n; i++) {
            if(n % i == 0) {
                num++;
                if(num == k) {
                    System.out.println(i);
                    break;
                }
            }
        }
        
        if(num < k) {
            System.out.println(0);
        }
    }
}

 

---

▷ num < k 인 경우를 밖으로 빼줘야 함