题目如下:
难度不是很大,就是对算法的优化有要求,不能用太无脑的方式。
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,这些排序算法和查找算法还得多写写,否则容易忘记。