728x90
반응형
문제 설명
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]
코드
처음에는
if(divisor == 1){
Arrays.sort(arr);
return arr;
}
List<Integer> list = new ArrayList();
for (int i = 0; i < arr.length; i++) {
if(arr[i]%divisor == 0){
list.add(arr[i]);
}
}
int listSize = list.size();
Integer[] result = list.toArray(new Integer[listSize]);
int[] answer = Arrays.stream(result).mapToInt(i -> i).toArray();
if(answer.length == 0){
answer = new int[1];
answer[0] = -1;
}
return answer;
이렇게 짜보았다.. 코드실행에서는 성공했는데 제출 채점에서는 실패..!
근데 아직 이유는 모르겠다.. 연구해봐야지..
다른 방법을 생각해서
import java.util.Arrays;
class Solution {
public int[] solution(int[] arr, int divisor) {
int count = 0;
for (int i = 0; i < arr.length; i++) {
if(arr[i]%divisor == 0){
count++;
}
}
int[] answer = new int[count];
count = 0;
for (int i = 0; i < arr.length; i++) {
if(arr[i]%divisor == 0){
answer[count]=arr[i];
count++;
}
}
Arrays.sort(answer);
if(answer.length == 0){
answer = new int[1];
answer[0] = -1;
}
return answer;
}
}
이 코드로 성공!
반응형
'프로그래머스 > 프로그래머스 - JAVA' 카테고리의 다른 글
[JAVA] 예산 (0) | 2023.02.23 |
---|---|
[JAVA] 시저 암호 (0) | 2023.02.23 |
[JAVA] 2016년 (0) | 2023.02.23 |
[JAVA] 최대공약수와 최소공배수 (0) | 2023.02.23 |
[JAVA] 신규 아이디 추천 (0) | 2023.02.23 |