搜索插入位置——leedcode35

搜索插入位置——leedcode35

给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。

请必须使用时间复杂度为 O(log n) 的算法。

---

思路：折半查找，基本上是实现折半查找，二分法

注意：要记录左右，注意设置的开闭，注意计算mid，终止条件要注意

答案

cpp

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int left, right;
        left = 0;
        right = nums.size();
        int mid;
        while(left < right){
            mid = left + (right - left)/2;
            if(nums[mid] < target){
                left = mid + 1;
            }
            else
                right = mid;
        }
        return left;
    }
};

---

python

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        left, right = 0, len(nums) #采用左闭右开区间[left,right)
        while left < right: # 右开所以不能有=,区间不存在
            mid = left + (right - left)//2 # 防止溢出, //表示整除
            if nums[mid] < target: # 中点小于目标值,在右侧,可以得到相等位置
                left = mid + 1 # 左闭,所以要+1
            else:
                right = mid # 右开,真正右端点为mid-1
        return left # 此算法结束时保证left = right,返回谁都一样