문제 설명
정수를 저장한 배열, arr 에서 가장 작은 수를 제거한 배열을 리턴하는 함수, solution을 완성해주세요. 단, 리턴하려는 배열이 빈 배열인 경우엔 배열에 -1을 채워 리턴하세요. 예를들어 arr이 [4,3,2,1]인 경우는 [4,3,2]를 리턴 하고, [10]면 [-1]을 리턴 합니다.
제한 조건
arr은 길이 1 이상인 배열입니다.
인덱스 i, j에 대해 i ≠ j이면 arr[i] ≠ arr[j] 입니다.
입출력 예
arr return
[4,3,2,1] [4,3,2]
[10] [-1]
class Solution {
public int[] solution(int[] arr) {
int[] answer = {};
if(arr.length==1){
answer = new int[] {-1};
}else{
int min = arr[0];
int index = 0;
answer = new int[arr.length-1];
for(int i=0; i<arr.length; i++){
if(arr[i]<min){
min = arr[i];
}
}
for(int i=0; i<arr.length; i++){
if(arr[i]!=min){
answer[index++] = arr[i];
}else{
continue;
}
}
}
return answer;
}
}
index++가 아닌 i로만 하고, continue를 써주지 않았더니 런타임 오류가 떴다.
조금 더 고민해봐야겠다.
import java.util.*;
import java.util.stream.Collectors;
class Solution {
public int[] solution(int[] arr) {
List<Integer> list = Arrays.stream(arr).boxed().collect(Collectors.toList());
Arrays.sort(arr);
list.remove(list.indexOf(arr[0]));
if (list.size() <= 0) return new int[]{-1};
return list.stream().mapToInt(i->i).toArray();
}
}
list를 활용할 수도 있군..