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

문제 설명

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

제한사항

• 1 ≤ n ≤ 10,000

입출력 예

입출력 예 설명

입출력 예 #1

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

입출력 예 #2

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

코드 설명

import java.util.ArrayList;
import java.util.Scanner;

public class Solution {
//	public ArrayList<Integer> solution(int n) { // 방식1
//		ArrayList<Integer> answer = new ArrayList<>(); // answer를 ArrayList로 해주고
//        for (int i = 1; i <= n; i++) {		
//        	if (n % i == 0) { // 만약 n % i == 0 이면(0이 나오면 약수이므로) answer에 i를 넣어줌.
//        		answer.add(i);
//        	}
//        }
//        return answer;
//    }
	public int[] solution(int n) { // 방식2
		ArrayList<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(); // 배열로 출력하기 위해 answer를 List에서 Int로 바꿔주고 배열로 출력해줌.
	}
	public static void main(String[] args) {
		Solution T = new Solution();
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
//		System.out.println(T.solution(n));
		for (int x : T.solution(n)) {
			System.out.print(x + " ");
		}
		sc.close();
	}
}