프로그래머스 - 약수 구하기

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