
public static int solution(int n) {
int sqrt = (int) Math.sqrt(n);
if (sqrt * sqrt == n) {
return 1;
} else {
return 2;
}
}
Math.sqrt(n)으로 제곱근 구하고
int로 형변환한 뒤 다시 제곱해서 n과 같으면 제곱수
√n, 즉 n의 제곱근을 double 타입으로 반환

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