[JAVA] 약수 구하기 - 배열크기

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

---

풀이 1)

class Solution {
    public int[] solution(int n) {
    	// answer 배열의 크기 구하기
        int sum = 0;
        for(int i=1; i<=n; i++){
            if(n%i==0){
                sum++;
            }
        }
        // answer 배열에 값 대입
        int[] answer = new int[sum];
        int j = 0;
        for(int i=1; i<=n; i++){
             if(n%i==0){
             answer[j]=i;
                 j++;
            }
        }
        return answer;
    }
}

 

풀이 2) 배열의 크기를 나중에 지정하는 방법

import java.util.List;
import java.util.ArrayList;

class Solution {
    public int[] solution(int n) {
        List<Integer> answer = new ArrayList<>();
        for(int i=1; i<=n; i++){
            if(n % i == 0){
                answer.add(i);
            }
        }
        return answer.stream().mapToInt(x -> x).toArray();
    }
}

List<타입> 이름 = new ArrayList<>()

값 추가할 때는 answer.add()

return 값은 이름. stream().mapToInt(x-> x).toArray();