【新手上路】LeetCode刷题之“搜索插入位置”

题目如下：

难度不是很大，就是对算法的优化有要求，不能用太无脑的方式。

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        if target in nums:
            return nums.index(target)
        else:
            nums.append(target)
            nums.sort()
            return nums.index(target)

 

但是抱着试试看的心态，我还是去尝试了最无脑的方法，分类讨论然后直接index方法，不知道是不是Cpython的算法优化比较好，居然通过了。 

c语言解题如下：

int searchInsert(int* nums, int numsSize, int target){
    for(int i=0;i<numsSize;i++)
    {
        if(nums[i]>=target) return i;  //如果找到了或者找到比target更大的数就返回索引
    }
    return numsSize;  //如果target比数组里任何树都大，那就返回数组长度，即target放在数组最后一项
}

 没想到就遍历一次内存消耗就这么拉跨了，不知道大佬们的精妙的算法是什么样的。

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        length = len(nums)
        for i in range(length):
            if nums[i]>=target:
                return i
        return length

 这是遍历法用Python解决的方案。

 遍历一次速度挺快，但是每次都得从第一个开始找，这样给定的数据一多就很麻烦了，二分查找可能是更好的方案。

int searchInsert(int* nums, int numsSize, int target){
    int low=0,high=numsSize-1;
    while(low<=high)
    {
        int mid=(low+high)/2;
        if(nums[mid]>target) high = mid-1;
        if(nums[mid]<target) low = mid+1;
        if(nums[mid]==target) return mid;
    }
    return low;
}

 

 果然二分查找yyds，这些排序算法和查找算法还得多写写，否则容易忘记。