본문 바로가기
코딩테스트/프로그래머스

배열 만들기 1

by 구너드 2023. 6. 9.

n 이하의 수 중에서 k의 배수들을 오름차순으로 저장하는 배열을 return 해야한다.

처음에는 n이하의 수 중 k로 나누었을 때 0인 숫자들만 ArrayList로 생성한 answer에 넣어주려고 했는데 곰곰이 생각하다보니까 int i의 초기값을 1로 설정하고 이후 조건식에서 i * k <= n이라는 식으로 구해주면 더 괜찮을 거라고 생각했다.

이를 코드로 작성하면

import java.util.*;

class Solution {
    public List solution(int n, int k) {
        List<Integer> answer = new ArrayList<>();
        for(int i = 1; k*i<=n;i++){
            answer.add(k*i);
        }
        return answer;
    }
}

와 같이 간단하게 표현할 수 있다.