搜索插入位置

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

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

示例：
输入: nums = [1,3,5,6], target = 7
输出: 4

核心思想是二分查找。

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

python我写的很复杂，应该可以更简单

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        l=len(nums)
        flag=0
        for i in range(0,l):
            if nums[i]==target:
                flag=1
                return i
        for j in range(0,l-1):
            if nums[j]<target and nums[j+1]>target:
                nums.insert(j+1,target)
                return j+1
        if target>nums[-1]:
            return l
        if target<nums[0]:
            return 0