[프로그래머스] 136798번 - 기사단원의 무기 [Java]

프로그래머스

 

 

---

 

 

문제

 

 

제출한 코드

import java.util.Arrays;

class Solution {
    public int solution(int number, int limit, int power) {
        
        int[] counts = new int[number];
        
        for (int i = 1; i <= number; i++) {
            int count = countDivisor(i);
            
            if (count > limit) counts[i-1] = power;
            else counts[i-1] = count;
        }
        
        return Arrays.stream(counts).sum();
    }

    private int countDivisor(int num) {
        double sqrt = Math.sqrt(num);
        int count = 1;
        
        for (int i = 2; i <= sqrt; i++) {
            if (num % i == 0) count++;
        }
        
        return sqrt == Math.floor(sqrt) ? count * 2 - 1 : count * 2;
    }
}

 

 

풀이

약수의 개수를 구하는 알고리즘은 아래의 글에 간단히 적어두었다.

약수의 개수 알고리즘

 

배열에 약수의 개수를 넣어서 마지막에 합을 구하여 return 하도록 작성했다.

약수의 개수를 구했을 때, 만약 limit 보다 더 많으면, 지정된 power를 counts 배열에 넣도록 했다.