💻 자바 알고리즘 - 배열 만들기 1

📕 문제 설명

정수 n과 k가 주어졌을 때, 1 이상 n이하의 정수 중에서 k의 배수를 오름차순으로 저장한 배열을 return 하는 solution 함수를 완성해 주세요.

 

📕 제한 사항

• 1 ≤ n ≤ 1,000,000
• 1 ≤ k ≤ min(1,000, n)

 

📝 답안

import java.util.*;

class Solution {
    public int[] solution(int n, int k) {
        ArrayList<Integer> answer = new ArrayList<>();
        for(int i=1; i<=n; i++) {
            if(i%k==0) {
                answer.add(i);
            }
        }

        return answer.stream().mapToInt(Integer::intValue).toArray();
    }
}

배열의 크기가 지정되지 않았을 땐 ArrayList를 이용해서 배열을 만들어주는 방식으로 풀어주면 된다.

이후 for문을 돌면서 리스트에 값을 넣어준 다음 리스트를 배열로 만들어서 리턴해준다.

 

📝 개선 답안

import java.util.*;

class Solution {
    public int[] solution(int n, int k) {
        int count = n/k;

        int[] answer = new int[count];
        int idx = 0;

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

        return answer;
    }
}

하지만 배열의 길이를 알 수 있다면 배열의 길이를 구한 후 배열에 넣고 리턴하는게 바람직하다.

1부터 n까지의 숫자중에서 k의 배수는 n/k와 동일하니 그렇게 배열의 길이 값을 구하여 배열을 초기화 한 뒤

원소를 넣어준다.

 

💡 참고

• 리스트.add() : 리스트 원소 추가