본문 바로가기
프로그래머스 Algorithm

[프로그래머스] 제일 작은 수 제거하기 Kotlin

by Echung 2023. 11. 21.

안녕하세요. 이번에는 프로그래머스 제일 작은 수 제거하기를 풀어보려고 합니다.

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr


Problem

정수를 저장한 배열, arr에서 가장 작은 수를 제거한 배열을 리턴하는 함수, solution을 완성해 주세요. 단, 리턴하려는 배열이 빈 배열인 경우엔 배열에 -1을 채워 리턴하세요. 예를 들어 arr이 [4,3,2,1]인 경우는 [4,3,2]를 리턴하고, [10] 면 [-1]을 리턴합니다.

 

[제한 사항]

arr은 길이 1 이상인 배열입니다.

인덱스 i, j 대해 i ≠ j이면 arr [i] ≠ arr [j]입니다.


사진 1. 입출력 예

Solution

class Solution {
    fun solution(arr: IntArray): IntArray {
        var answer = intArrayOf()
        var min = Int.MAX_VALUE
        
        if(arr.size == 1) {
            answer += -1
            return answer
        }
        
        for(i in 0..arr.size - 1) {
            if(min > arr[i]) {
                min = arr[i]    
            }    
        }
        
        for(i in 0..arr.size - 1) {
            if(min != arr[i]) {
                answer += arr[i]
            }
        }
        
        return answer
    }
}

 이번 문제는 for문을 통해서 문제에 접근해 보았다. 

 

1. 다른 사람 코드

class Solution {
    fun solution(arr: IntArray): IntArray {        
    
        if(arr.size == 1) return intArrayOf(-1)
        val answer = arr.toMutableList()
        answer.remove(answer.minOf { it })
        return answer.toIntArray()
    }
}

  다른 사람의 코드를 통해서 더 나은 방법을 공부 중에 minOf()라는 함수를 알게 되었다. minOf를 통해서 제일 작은 값을 찾고 제거해 주는 방식의 코드가 확실히 가독성면에서는 더 좋아 보인다. 


Performance

1. 내가 작성한 코드 2. 다른 사람 코드
반응형