[프로그래머스] 기사단원의 무기

문제

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

프로그래머스

풀이와 코드

사실상 약수의 개수를 구하는 알고리즘이 중요한 문제!

좀 전의 소수의 개수를 구하는 문제를 풀어봐서 이번 문제 풀 때 도움이 됐다ㅎㅎ

기억하자... 제곱... 효울성 증대....

#include <string>
#include <vector>
#include <iostream>

using namespace std;

int solution(int number, int limit, int power) {
    int answer = 0;
    
    for(int i = 0; i < number; i++){
        // 약수의 개수
        int count = 0;
        for(int j = 1; j * j <= i + 1; j++){
            if ((i + 1) % j == 0){
                if (j * j == (i + 1)){
                    count++;
                }else{
                    count += 2;
                }
            }
        }
        
        // 협약으로 정해진 제한 수치 확인
        if (count > limit){
            answer += power;
        }else{
            answer += count;
        }
    }
    
    return answer;
}