[Java] 알고리즘, "나누어 떨어지는 숫자 배열"
- 알고리즘 문제/[Java] 알고리즘
- 2019. 2. 3. 14:57
나누어 떨어지는 숫자 배열
문제 설명
array의 각 element 중 divisor로 나누어 떨어지는 값을 오름차순으로 정렬한 배열을 반환하는 함수, solution을 작성해주세요.
divisor로 나누어 떨어지는 element가 하나도 없다면 배열에 -1을 담아 반환하세요.
제한사항
arr은 자연수를 담은 배열입니다.
정수 i, j에 대해 i ≠ j 이면 arr[i] ≠ arr[j] 입니다.
divisor는 자연수입니다.
array는 길이 1 이상인 배열입니다.
입출력 예
배열 |
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 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | import java.util.*; class Solution { public int[] solution(int[] arr, int divisor) { int[] answer = {}; ArrayList<Integer> temp = new ArrayList<Integer>(); for(int i =0; i < arr.length; i++){ if(arr[i]%divisor==0){ temp.add(arr[i]); } } if(temp.size()==0){ answer = new int[1]; answer[0]=-1; }else{ answer = new int[temp.size()]; for(int i =0; i<temp.size(); i++){ answer[i]=temp.get(i); } Arrays.sort(answer); } return answer; } } | cs |
일단 먼저 순수배열은 다루기에 굉장히 유연하지 못한 도구입니다. 선언하는 배열의 공간조차 직접 10이든 100이든 선언해야 하죠. 따라서 임시배열로써 temp라는 ArrayList를 선언하여 길이에 유연하도록 하였습니다. 그 다음에 for문으로 배열을 돌고 divisor로 나누면서 나머지가 0인 친구들을 temp임시배열에 저장하고, temp의 길이가 0이면 divisor로 나누어 떨어지는 수가 없으므로 answer배열에 1의 공간을 만들고 거기에 -1을 대입합니다.
temp배열의 길이가 0보다 크다면(즉, 나누어 떨어지는 수가 존재한다면) temp의 길이만큼 answer배열의 공간을 만들어주고 일단 temp의 내용물을 직접복사합니다. 그리고나서 정렬을 해주면 되는 것입니다.!!
다른사람의 풀이
1 2 3 4 5 6 7 8 9 10 11 12 13 14 | 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) )); } } | cs |
stream과 Lambda를 이용하신분도 계시고...(참고로 저는 스트림과 람다는 그다지 좋아하는 코딩스타일이 아닙니다. 너무 어려워서ㅠㅠ)
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | import java.util.Arrays; class Divisible { public int[] divisible(int[] array, int divisor) { //ret에 array에 포함된 정수중, divisor로 나누어 떨어지는 숫자를 순서대로 넣으세요. int count=0,two=0; for(int i=0; i< array.length; i++){ if(array[i]%divisor==0) count++; } int[] ret = new int[count]; for(int i=0; i< array.length; i++){ if(array[i]%divisor==0) ret[two++]=array[i]; } 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) )); } } | cs |
이분은 for문을 돌때 count를 세서 반환할 배열의 길이를 구하셨네요. 자바 라이브러리를 이용하지 않으시고 순수배열로서 처리하신 점이 되게 존경스럽습니다 ㅎㅎ. 아직까지 저 two 변수의 활용이 잘 이해되진 않네요 ㅠㅠ 아시는분은 댓글로 남겨주세요 ㅠㅠ 한수배우도록하겠습니다.!!
'알고리즘 문제 > [Java] 알고리즘' 카테고리의 다른 글
[Java] 자바 알고리즘, "문자열 내 p와 y의 개수" (0) | 2019.02.06 |
---|---|
[Java] 자바 알고리즘, "두 정수 사이의 합" (0) | 2019.02.06 |
[Java] 알고리즘, "같은 숫자는 싫어" (1) | 2019.02.02 |
[Java] 알고리즘, "가운데 글자 가져오기" (0) | 2019.02.02 |
[Java] 알고리즘, "2016년 a월 b일을 맞춰라" (0) | 2019.02.01 |