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

문제 링크

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

프로그래머스

문제 설명

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

제한사항

• 1 ≤ n ≤ 10,000

입출력 예

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

입출력 예 설명

입출력 예 #1

• 24의 약수를 오름차순으로 담은 배열 [1, 2, 3, 4, 6, 8, 12, 24]를 return합니다.

입출력 예 #2

• 29의 약수를 오름차순으로 담은 배열 [1, 29]를 return합니다.

나의 풀이

Java

class Solution {
    public int[] solution(int n) {
        int count = 0;
        for(int i = 1; i <= n; i++) {
            if(n%i == 0) count++;
        }
        
        int[] answer = new int[count];
        for(int i = 1, j = 0; i <= n; i++) {
            if(n%i == 0) {
                answer[j] = i;
                j++;
            }
        }
        
        return answer;
    }
}

JavaScript

function solution(n) {
    let answer = [];
    for(let i = 1; i <= n; i++) {
        if(n%i == 0) answer.push(i);
    }
    return answer;
}