알고리즘/프로그래머스

코딩테스트 연습 / 연습문제 / 제일 작은 수 제거하기

외계공룡 2020. 4. 14. 18:35

문제 설명

정수를 저장한 배열, 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 answer = new int[arr.length-1];
		
	  int min = 999999;
	  for(int i=0; i<arr.length; i++) {
		  if(min > arr[i]) min = arr[i];
	  }
		
	  int j = 0;
	  for(int i=0; i<arr.length; i++) {
		  if(min == arr[i]) continue;
		  answer[j++] = arr[i];
	  }	
	  return answer;
  }
}

풀이 분석

문제를 풀때 처음에 정렬을 하려고 했다. 하지만 정렬시 기존 배열의 순서가 섞이는 문제가 생겨서 min변수를 활용하기로 했다. 다른 사람들은 리스트로 변형해서 제거를 하거나 스트림을 이용한 사람도 많았다. 혹시나 배열을 자를 수 있는 메소드가 있나 싶어서 몇번 더 찾아보았지만 그런 메소드는 없는 것 같아보인다. 가장 인상깊게 본 것은 min변수에 나처럼 높은 숫자를 넣는 것이 아닌 arr[0]으로 초기화 한것이다. 배열의 모든 원소가 다행이 999999보다 낮은 숫자가 없어서 문제를 맞았지만 다음에는 arr[0]을 쓰는 것이 좋을 것 같다.

 

추가로 복습하면서 리스트로 만들 수 있지 않을까 생각해보았는데 remove메소드는 매개변수로 인덱스를 받기 때문에 불가능 하다는 것을 깨달았다.