LeetCode：35.搜索插入位置

35.搜索插入位置

• 给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。
• 请必须使用时间复杂度为 $O(logn)$ 的算法。

---

方法一：二分法

因为题目要求时间复杂度为$O(logn)$，所以考虑使用 二分法 在数组中进行搜索。二分法的思想是逐步搜小搜索区间。因为数组中的数组是有序的，我们先比较数组中间的元素与待查找元素的大小，根据判断结果可以判定：

• 当前元素等于待查找元素，返回当前元素的下标；

• 当前元素小于待查找元素，则下一步我们在当前元素的右侧数组中进行查找；

• 当前元素大于待查找元素，则下一步我们在当前元素的左侧数组中进行查找；

int searchInsert(vector<int>& nums, int target) {
        if (nums.size() == 0) return 0;
        //nLeft、nRight分别指向当前查找数组的边界
        int nLeft = 0;
        int nRight = nums.size();
        //nPos 用于定位 当前查找数组的中间
        int nPos = (nLeft + nRight) / 2;

        while (nPos != nLeft && nPos!=nRight) {
            if (nums[nPos] == target) return nPos;
            else if (nums[nPos] < target) {
                //查找 nPos 右侧的数组
                nLeft = nPos;
                nPos = (nLeft + nRight) / 2;
            }
            else {
                //查找 nPos 左侧的数组
                nRight = nPos;
                nPos = (nLeft + nRight) / 2;
            }

        }
        //当中间位置和边界重合时跳出循环，还需做一次判断
        return nums[nPos] < target ? nPos + 1 : nPos;
        /*if (nums[nPos] == target) return nPos;
        else if (nums[nPos] < target) return nPos + 1;
        else { return nPos; }*/
    }

时间复杂度：$O(\log n)$，其中 $n$ 为数组的长度。二分查找所需的时间复杂度为 $O(\log n)$。

空间复杂度：$O(1)$

---

LeetCode题解

class Solution {
    public int searchInsert(int[] nums, int target) {
        int n = nums.length;
        int left = 0, right = n - 1, ans = n;
        while (left <= right) {
            int mid = ((right - left) >> 1) + left;
            if (target <= nums[mid]) {
                ans = mid;
                right = mid - 1;
            } else {
                left = mid + 1;
            }
        }
        return ans;
    }
}

因为带查找的左右边界一直在变化，所以循环的判断条件可以设为 left <= right 。判断条件的设定，带查找区间的开闭可以根据自己的习惯来设置，设置不同可能要注意返回值不同。只要掌握 二分法 的思想即可。