[프로그래머스] Lv0 약수 구하기

문제 설명

정수 n이 매개변수로 주어질 때, n의 약수를 오름차순으로 담은 배열을 return하도록 solution 함수를 완성해주세요.

 

제한사항

1 <= n <= 10,000

 

입출력 예

n	result
24	[1, 2, 3, 4, 6, 8, 12, 24]
29	[1, 29]

 

풀이

import java.util.ArrayList;
import java.util.Comparator;

class Solution {
    public ArrayList<Integer> solution(int n) {
        int sqrt = (int) Math.sqrt(n);
        ArrayList<Integer> arr = new ArrayList<>();

        for (int i = 1; i <= sqrt; i++) {
            if (n % i == 0) {
                arr.add(i);
            
            if (n / i != i) {
                arr.add(n / i);
            }
        }
    }       
    arr.sort(Comparator.naturalOrder()); 
    return arr;
    }
}

주어진 숫자 n 이하의 모든 숫자를 if문으로 검증하면서 약수인지 판단하는 방법도 있지만 이 경우 n값이 커지면 커질수록 효율성이 매우 떨어진다.

 

그렇기 때문에 Math.sqrt 를 활용해 반복해야 하는 범위를 줄였다. (스퀘어 함수를 사용하면 n = 루트n * 루트n으로 표시할 수 잇는데 1부터 루트 n까지 범위를 기준으로 해당 숫자를 나누고 떨어진다면 그 숫자가 약수가 되는 것을 알 수 있다.)