프로그래머스-level1-나누어 떨어지는 숫자 배열

https://programmers.co.kr/learn/courses/30/lessons/12910

코딩테스트 연습 - 나누어 떨어지는 숫자 배열 | 프로그래머스

나의 코드

Arrays.sort를 활용해서 정렬을 먼저 하였고

divisor로 나누어떨어지는 것을 배열에 저장하였다.

최종적으로 값이 0이 아닌 값들은 제외하고 배열을 새로 만들어 반환함.

나누어지는게 하나도 없는 경우는 boolean을 활용해서 -1 반환.

import java.util.Arrays;
class Solution {
  public int[] solution(int[] arr, int divisor) {
      int[] answer = new int[arr.length];
	      boolean check = false;
	      Arrays.sort(arr);
	      int cnt = 0;
	      for(int i=0; i< arr.length; i++) {
	    	  if(arr[i]%divisor == 0) {
	    		  check = true;
	    		  answer[cnt] = arr[i];
	    		  cnt++;
	    	  }
	      }
	      int length = 0;
	      for(int i =0; i < answer.length; i++) {
	    	  if(answer[i] == 0) {
	    		  break;
	    	  }
	    	  length++;
	      }
	      
	      
	      if(check == false) {
	    	  return new int[] {-1};
	      }
	      return Arrays.copyOfRange(answer, 0, length);
  }
}

 

참고할 코드

한줄이면 된다..

Arrays.stream에 대해서 공부가 필요할듯!