C++ / 프로그래머스 / 기사단원의 무기

문제

https://school.programmers.co.kr/learn/courses/30/lessons/136798

프로그래머스

문제풀이

이중 for문을 이용해 약수를 구한다.

약수를 구하고 약수의 개수가 담긴 divisor벡터를 탐색하면서 limit를 넘긴 값이 있으면 power로 바꾼다.

divisor벡터의 모든 값을 더하고 return 한다.

 

코드

#include <string>
#include <vector>

using namespace std;

int solution(int number, int limit, int power) {
    int answer(0), idx(0), idx2(0);
    vector<int>divisor(number + 1, 1);
    for (idx = 2; idx <= number; ++idx) {
        for (idx2 = 1; idx * idx2 <= number; ++idx2) {
            divisor[idx * idx2]++;
        }
    }
    
    for (idx = 1; idx <= number; ++idx) {
        if (divisor[idx] > limit)
            divisor[idx] = power;
        answer += divisor[idx];
    }

    return answer;
}