백준 2501번 C++

백준 2501번 약수

https://www.acmicpc.net/problem/2501

2501번: 약수 구하기

정답 코드

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
vector<int>arr;
int main() {
	int n, k;
	cin >> n >> k;
	int index = 0;
	for (int x = 1; x <= n; x++) {
		if (n % x == 0) {
			arr.push_back(x);
		}
	}
	sort(arr.begin(), arr.end());
	int d = k - 1;
	if (arr.size() >= k) {
		cout << arr[d];
	}
	else
		cout << 0;

}

문제 풀이

약수를 구하는 문제입니다. 약수는 for문을 돌려서 구한 후 벡터에 삽입하였습니다. 그리고 algorithm헤더파일에 있는 sort함수를 이용해서 정렬하였습니다. 마지막으로 벡터의 크기가 k개보다 작으면 0을 그게 아니라면 k번째 수를 출력하도록 조건문을 추가했습니다.