[프로그래머스] Level 1 – 나누어 떨어지는 숫자 배열

문제 설명

 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]	5	[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]을 리턴합니다.

---

- 풀이 과정 -

1.     배열arr의 길이만큼 반복문을 수행하면서 배열의 원소 하나씩 변수 divisor의 값으로 나누어 떨어지는지 확인

2.     나누어 떨어지는 값이 있을 경우 arrlist에 저장

3.     arrlist에 저장된 값이 없는 경우 나누어 떨어지는 값이 없는 것으로 answer 배열에 -1의 값으로 초기화 후 리턴

4.     arrlist에 저장된 값이 있는 경우 answer 배열에 복사하여 정렬 후 리턴

---

 - 풀이 코드1 –

import java.util.ArrayList;
import java.util.Arrays;
 
class Solution {
    public int[] solution(int[] arr, int divisor) {
        ArrayList<Integer> arrlist = new ArrayList<Integer>();
        
        for(int i=0;i<arr.length;i++) {
        if(arr[i]%divisor==0) {
            arrlist.add(arr[i]);
         }
       }
        
        if(arrlist.size()==0) {
            int[] answer = {-1};
            return answer;
        }
        
        int[] answer = new int[arrlist.size()];
        for(int i=0;i<arrlist.size();i++) {
            answer[i]=arrlist.get(i);
        }
        Arrays.sort(answer);
        return answer;
    }
    
    public static void main(String[] args) {
        Solution sol = new Solution();
        int[] arr  = {3,2,6};
        int divisor = 10;
        System.out.println(Arrays.toString(sol.solution(arr,divisor)));
    }
}

 

- 풀이 코드2 –

import java.util.Arrays;
 
class Solution {
    public int[] solution(int[] arr, int divisor) {
        int count = 0,two=0;
        for(int i=0;i<arr.length;i++) {
            if(arr[i]%divisor==0) count++;
        }
        
        if(count==0) {
            int[] answer = {-1};
            return answer;
        }
        
        int[] answer = new int[count];
        for(int i=0;i<arr.length;i++) {
            if(arr[i]%divisor==0) answer[two++]=arr[i];
        }
        Arrays.sort(answer);
        return answer;
    }
    
    public static void main(String[] args) {
        Solution sol = new Solution();
        int[] arr  = {2,36,1,3};
        int divisor = 1;
        System.out.println(Arrays.toString(sol.solution(arr,divisor)));
    }
}