알고리즘

[프로그래머스] 약수의 갯수와 덧셈

winwin-k9 2023. 8. 16. 17:12

문제

두 정수 left와 right가 매개변수로 주어집니다.

left부터 right까지의 모든 수들 중에서, 약수의 개수가 짝수인 수는 더하고, 약수의 개수가 홀수인 수는 뺀 수를 return 하도록 solution 함수를 완성해주세요.

풀이

가장 간단하게 생각한 것은 어떤 수가 주어지면 이를 1부터 n까지 소수를 count시키고
문제에서 주어진대로 더하거나 빼면 된다.
통과는 했지만, 결국 n번 반복을 또 돌어야 하기 때문에 매우 비효율적이다.
다른 사람이 푼걸 보니까 제곱수를 이용한 풀이가 있더라.
제곱수로 나누어지면 홀수개이고, 아니면 짝수개이므로 약수 갯수 검사가 필요가 없다.
세상엔 천재가 많은듯...


import java.util.*;
import java.lang.*;

class Solution {
    public int solution(int left, int right) {
        int answer = 0;
        for(int i = left; i <= right; i++) {
            int count = getDivisorCount(i);
            if (count % 2 == 0) {
                answer += i;
            } else {
                answer -= i;
            }
        }
        return answer;
    }

    static int getDivisorCount(int num) {
        int count = 0;
        for(int i = 1; i <= num; i++) {
            if (num % i == 0) {
                count++;
            }
        }

        return count;
    }
}

https://github.com/Win-9/Algorism/tree/main/programers/%EC%95%BD%EC%88%98%EC%9D%98%20%EA%B0%9C%EC%88%98%EC%99%80%20%EB%8D%A7%EC%85%88

728x90