level 1. 나누어 떨어지는 숫자 배열 (java)
11 Jun 2017 | 알고리즘 프로그래밍문제
divisible 메소드는 int형 배열 array와 int divisor를 매개변수로 받습니다. array의 각 element 중 divisor로 나누어 떨어지는 값만 포함하는 새로운 배열을 만들어서 반환하도록 divisible에 코드를 작성해 보세요.
예를들어 array가 {5, 9, 7, 10}이고 divisor가 5이면 {5, 10}을 리턴해야 합니다.
풀이코드
- ArrayList 에 조건에 맞는 값을 추가하고, 이를 다시 Array 로 변경하여 리턴하였다.
- ArrayList
는 .toArray() 를 통해서 int[] 배열로 쉽게 변경되지 않는다. (Integer, int 의 차이 때문에) - 따라서 ArrayList 내부의 모든 element 를 int로 형변환 해야한다.
이때 stream의 mapToInt 메소드를 활용하면 간단하게 처리 가능하다.
import java.util.Arrays;
import java.util.ArrayList;
class Divisible {
public int[] divisible(int[] array, int divisor) {
ArrayList<Integer> divArray = new ArrayList<Integer>();
for(int i : array){
if(i % divisor == 0){
divArray.add(i);
}
}
int[] ret = new int[divArray.size()];
for(int i=0; i<ret.length; 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.ArrayList;
import java.util.Arrays;
public class Divisible {
public int[] divisible(int[] array, int divisor) {
ArrayList<Integer> ret = new ArrayList<>();
for (int i = 0; i < array.length; i++) {
if (array[i] % divisor == 0) {
ret.add(array[i]);
}
}
return ret.stream().mapToInt(i -> i).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)));
}
}
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) ));
}
}
배운점
- ArrayList 는 .add() 메소드를 통해서 요소를 추가할 수 있지만, Array는 array[0] 과 같이 index를 사용해야 한다.
- Arrays.toString() 를 통해서 배열을 그대로 print 할 수 있다. (Returns a string representation of the contents of the specified array.)
- .stream() 을 통해서 collections 를 좀 더 편리하게 처리할 수 있다. (API 문서, 참고 글)