알고리즘/프로그래머스
[Level 1] 나누어 떨어지는 숫자 배열
Dev_YooJin
2018. 1. 15. 21:02
출저 https://programmers.co.kr/learn/challenge_codes/89
문제
divisible 메소드는 int형 배열 array와 int divisor를 매개변수로 받습니다. array의 각 element 중 divisor로 나누어 떨어지는 값만 포함하는 새로운 배열을 만들어서 반환하도록 divisible에 코드를 작성해 보세요. 예를들어 array가 {5, 9, 7, 10}이고 divisor가 5이면 {5, 10}을 리턴해야 합니다.
풀이
import java.util.Arrays;
import java.util.ArrayList;
class Divisible {
public int[] divisible(int[] array, int divisor) {
ArrayList divArray = new ArrayList();
//ret에 array에 포함된 정수중, divisor로 나누어 떨어지는 숫자를 순서대로 넣으세요.
int arrLen = array.length;
for (int i = 0; i < arrLen; i++) {
if (array[i] % divisor == 0) {
divArray.add(array[i]);
}
}
int[] ret = new int[divArray.size()];
int retLen = ret.length;
for(int i=0; i < retLen; i++) {
ret[i] = divArray.get(i).intValue();
}
return ret;
}
public static void main(String[] args) {
Divisible div = new Divisible();
int[] array = {5, 9, 7, 10};
System.out.println( Arrays.toString ( div.divisible(array, 5) ));
}
}
다른 사람들의 풀이
import java.util.Arrays;
class Divisible {
public int[] divisible(int[] array, int divisor) {
//ret에 array에 포함된 정수중, divisor로 나누어 떨어지는 숫자를 순서대로 넣으세요.
return Arrays.stream(array).filter(factor -> factor % divisor == 0).toArray();
}
// 아래는 테스트로 출력해 보기 위한 코드입니다.
public static void main(String[] args) {
Divisible div = new Divisible();
int[] array = {5, 9, 7, 10};
System.out.println( Arrays.toString( div.divisible(array, 5) ));
}
}