문제 설명
정수 n과 정수 배열 numlist가 매개변수로 주어질 때, numlist에서 n의 배수가 아닌 수들을 제거한 배열을 return하도록 solution 함수를 완성해주세요.
제한사항
1 ≤ n ≤ 10,000 1 ≤ numlist의 크기 ≤ 100 1 ≤ numlist의 원소 ≤ 100,000
나의 코드
import java.util.List;
import java.util.ArrayList;
class Solution {
public int[] solution(int n, int[] numlist) {
List<Integer> list = new ArrayList<>();
for(int i=0; i<numlist.length; i++) {
if(numlist[i]%n==0) list.add(numlist[i]);
}
int[] answer = new int[list.size()];
for(int i=0; i<list.size(); i++) {
answer[i] = list.get(i);
}
return answer;
}
}
다른 사람 코드
class Solution {
public int[] solution(int n, int[] numlist) {
int count = 0;
for(int i : numlist){
if(i%n==0){
count++;
}
}
int[] answer = new int[count];
int idx = 0;
for(int i : numlist){
if(i%n==0){
answer[idx]=i;
idx++;
}
}
return answer;
}
}
느낀 점
n으로 나눈 나머지가 0일 때 새로운 리스트에 추가될 수 있도록 하여 어렵지 않게 풀어냈다.
다른 사람 코드와 같이 변수를 이용하여 풀면은 굳이 List를 사용하지 않고도 풀 수 있었다!