[C/C++] 백준 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 이하이다.

ex)

6 3

25 4

2735 1

 

 

출력

 

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

ex)

3

0

1

 

---

---

 

코드

 

#include<iostream>
#include <vector>
#include <algorithm> // sort()

using namespace std;

class exam {

public:
	void test(int n, int k) {
		
		vector<int> num;

		for (int i = 1; i <= n; i++) {

			if (n % i == 0) {
				num.push_back(i);
			}	
		}

		sort(num.begin(), num.end()); // 오름차순 정렬

		if (num.size() < k) {

			cout << 0;
		}
		else {

			cout << num[k - 1];
		}
	}
};

int main()
{
	int N,K;

	cin >> N>>K;

	exam ex = exam();

	ex.test(N,K);

	return 0;
}