알고리즘/프로그래머스

[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) ));
    }
}