[java] 2501. 약수 구하기

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

public class Main {
    public static void main(String[] args) throws IOException {
        //첫째 줄에 자연수 n과 k가 공백을 두고 입력. k는 1이상 n이하
        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[] arr = new int[n];
        int count = 0;

        //n의 약수구하기 + 작은거 저장 >> 어차피 오름차순 정렬인데 k번째 그냥 구해도 되지 않을까
        for (int i = 1; i <= n; i++) {
            if (n % i == 0) {
                arr[count] = i;
                count++;
            }
        }
        Arrays.sort(arr,0,count);	//오름차순 정렬

        if (k > count) {
            System.out.println("0");
        } else {    //n의 약수들 중 k번째 작은 수 출력
            System.out.println(arr[k-1]);
        }
    }


}

count를 이용해서 array 인덱스를 설정하는 방법을 알았다!