[백준 풀이_Java] 2501 약수구하기

문제

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

 

 

 

처음 생각

 

약수를 구해서 리스트에 넣고(배열은 고정적이라서 불편하다고 판단)

오름차순으로 정렬해서 K번째 수(리스트에서는 K-1 번째)를 구한다

 

 

 

내가 푼 방법

 

import java.util.ArrayList;
import java.util.Collections;
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();
        // 리스트생성
		ArrayList<Integer> arr = new ArrayList<>();
		
        // 리스트에 약수넣기
		for(int i=1;i<=N;i++) {
			if(N%i ==0) arr.add(i);
		}
		
        // 오름차순
		Collections.sort(arr);
        
        // 출력
		if(K>arr.size()) System.out.println(0);
		else System.out.println(arr.get(K-1));
	}

}