lv0_코딩테스트 입문_소인수분해

코딩테스트 연습 - 소인수분해 | 프로그래머스 스쿨 (programmers.co.kr)

프로그래머스

✔️ 유형 : STL(algoritm)

✔️ 문제 풀이: 소인수분해를 한뒤, unique() 함수를 이용하여 중복값을 제거 한 후, erase() 함수를 이용해서 벡터 뒷부분의 쓰레기 값을 제거한다.

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

vector<int> solution(int n) {
    vector<int> answer;
    int target = n;
    int prime = 2;
    while(target != 1)
    {
        if(target%prime == 0)
        {
            answer.push_back(prime);
            target /= prime;
        }
        else
            prime++;
    }
    
    sort(answer.begin(), answer.end());
    
    answer.erase(unique(answer.begin(), answer.end()), answer.end());
    return answer;
}