[JAVA/알고리즘] 프로그래머스 나누어 떨어지는 숫자 배열

문제

---

array의 각 element 중 divisor로 나누어 떨어지는 값을 오름차순으로 정렬한 배열을 반환하는 함수, solution을 작성해주세요. divisor로 나누어 떨어지는 element가 하나도 없다면 배열에 -1을 담아 반환하세요.

 

제한사항

• arr은 자연수를 담은 배열입니다.
• 정수 i, j에 대해 i ≠ j 이면 arr[i] ≠ arr[j] 입니다.
• divisor는 자연수입니다.
• array는 길이 1 이상인 배열입니다.

입출력 예

---

arr	divisor	return
[ 5, 9, 7, 10 ]	10	[ 5, 10 ]
[ 2, 36, 1, 3 ]	1	[ 1, 2, 3, 36 ]
[ 3, 2, 6 ]	10	[ -1 ]

입출력 예 설명

---

입출력 예#1
arr의 원소 중 5로 나누어 떨어지는 원소는 5와 10입니다. 따라서 [5, 10]을 리턴합니다.

 

입출력 예#2
arr의 모든 원소는 1으로 나누어 떨어집니다. 원소를 오름차순으로 정렬해 [1, 2, 3, 36]을 리턴합니다.

 

입출력 예#3
3, 2, 6은 10으로 나누어 떨어지지 않습니다. 나누어 떨어지는 원소가 없으므로 [-1]을 리턴합니다.

나의 풀이

---

class Solution {
  public int[] solution(int[] arr, int divisor) {
      int counter = 0;
      int sortArr[] = new int[arr.length];
      for (int i = 0; i < arr.length; i++) {
          if (arr[i] % divisor == 0) {
              sortArr[counter++] = arr[i];
          }
      }
        
      if (counter != 0) {
        int[] sortArr2 = new int[counter];
        System.arraycopy(sortArr, 0, sortArr2, 0, counter);
            for (int i = 0; i < sortArr2.length; i++) {
                for (int j = 0; j < i; j++) {
                    if (sortArr2[i] < sortArr2[j]) {
                        int temp = sortArr2[i];
                        sortArr2[i] = sortArr2[j];
                        sortArr2[j] = temp;
                    }
                }
            }
        return sortArr2; 
        }else {
             int[] sortArr2 = {-1};
            return sortArr2;
        }
  	 }
  }

전체 파라미터로 넘어온 arr의 배열만큼 탐색을 위해 반복문을 arr의 크기만큼 반복문을 돌리고 주어진 나눔수로 나눠지는 지 체크해서 임시배열인 sortArr에 값을 저장한다. 그 후 System.arraycopy를 통해 임시 저장된 sortArr을 복사 후 오름 차순으로 값 정렬 진행 후 리턴 없을 경우 { - 1 } 배열을 리턴한다.

 

---

마치며

 

다른 사람의 풀이는 람다식을 통해 arr을 추출하고 Arrays.sort를 통해 정렬을 사용한 방식을 선택했다. 현재 개발 환경상 예전 버전의 자바를 사용하고 있어서 람다식을 사용할일이 없다는 핑계로 있다는 것만 알고 알아보지는 않았는데 한번 공부를 해야겠다는 생각이 드는 간결한 코드였다.

 

다른 사람의 풀이

class Solution {
  public int[] solution(int[] arr, int divisor) {
          int[] answer = Arrays.stream(arr).filter(factor -> factor % divisor == 0).toArray();
          if(answer.length == 0) answer = new int[] {-1};
          java.util.Arrays.sort(answer);
          return answer;
  }
}