題目連結

題型解說

這是一題難度為簡單的題目

需要設計一個方法，此方法會傳入一個排序好的整數陣列 nums 和一個目標數字 target

找出 target 在 nums 的索引返回，如果沒有找到，則返回插入的索引

範例:

nums = [1,3,5,6], target = 5，則回傳 2

nums = [1,3,5,6], target = 2，則回傳 1，因為 2 不存在 nums 中，又因 2 小於 3，所以插入時會在 3 的位置，所以回傳 3 的索引 1

解題思路

這一題其實不困難，只要一個迴圈找過一次就能順利過關

判斷一下當前元素是否大於等於目標元素，如果是就回傳當前索引

若迴圈執行完畢表示目標元素比 nums 的所有元素都大，那就回傳 nums 的長度

只是這題大多數的人都是用 二分搜尋法 來解題

所以兩種解法都附上給大家做參考

程式碼

Java

class Solution {
    public int searchInsert(int[] nums, int target) {
        int len = nums.length;
        
        for (int i = 0; i < len; i++) {
            if (nums[i] >= target) {
                return i;
            }
        }
        
        return len;
    }
}

class Solution {
    public int searchInsert(int[] nums, int target) {
        int left = 0; 
        int right = nums.length - 1; 

        while(left <= right) { 
            int mid = (left + right) / 2; 
            
            if(nums[mid] < target) 
                left = mid + 1; 
            else if(nums[mid] > target) 
                right = mid - 1; 
            else 
                return mid; 
        } 

        return left; // 因為題目要求找不到元素時得回傳插入的索引，不然正常情況應是回傳 -1
    }
}