[ LeetCode] 35 Search Insert Position

📌 Problem

nums 배열은 서로 다른 정수가 정렬된 상태인 배열이다. 주어진 target 정수가 몇 번째 index에 있는지를 반환하면 된다. 만약 target 정수가 nums 배열에 없다면 몇 번째 index에 삽입해야 nums의 특성이 유지되는지를 반환하면 된다. 예를 들어 num = [1, 3, 5, 6]이고 target = 2이면 2가 nums에 없기 때문에 1과 3 사이에 삽입하면 된다. 결과적으로 1을 반환하면 된다.

📌 Solution

nums 배열을 순서대로 탐색하면서 target과 비교하면 된다. 만약 현재 탐색한 값이 target보다 크거나 같으면 현재 index를 반환하면 된다. 이때 마지막까지 탐색을 완료한 후를 고려해야 한다. 이때는 nums의 크기를 반환하면 된다.

📌 Code

fun searchInsert(nums: IntArray, target: Int): Int = nums.mapIndexed { idx, num ->
    if (num >= target) return idx
}.size