[알고리즘문제풀기] 약수 구하기

문제

내가 작성한 정답

ArrayList → Array

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

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

	//streamAPI사용하지 않고 ArrayList -> Array로 변환
	
        int[] answer = new int[a.size()];
        for(int i = 0; i<a.size(); i++){
            answer[i] = a.get(i);
        }
        return answer;
    }
}

StreamAPI

import java.util.stream.IntStream;

class Solution {
    public int[] solution(int n) {
        return IntStream.rangeClosed(1, n) 
                .filter(i -> n%i == 0) 
                .toArray(); 
    }
}

다른 사람들이 작성한 정답

ArrayList 사용하지 않음

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

        answer = new int[cnt];

        cnt = 0;
        for (int i = 1; i <= n; i++) {
            if (n % i == 0) {
                answer[cnt] = i;
                cnt++;
            }
        }

        return answer;
    }
}