35. Search Insert Position(python+cpp)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_21275321/article/details/83507002

题目:

Given a sorted array and a target value, return the index if the target is found. If not, return the index where it would be if it were inserted in order.
You may assume no duplicates in the array.
Example 1:

Input: [1,3,5,6], 5 
Output: 2

Example 2:

Input: [1,3,5,6], 2 
Output: 1 

Example 3:

Input: [1,3,5,6], 7 
Output: 4

Example 4:

Input: [1,3,5,6], 0 
Output: 0

解释:
二分查找问题,事实上就是实现bisct.bisct_left(),注意不是bisect.bisect(),因为如果数字在数组中已经出现了的话bisect.bisect()返回的是可以插入的位置,有可能返回的是数字出现的位置的后一个位置,但是这里题目要求返回的是数字的位置。(即数字在数组中第一次出现的位置)
用库函数实现:
python代码:

import bisect
class Solution(object):
    def searchInsert(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: int
        """
        return bisect.bisect_left(nums,target)

c++代码:

#include<algorithm>
using namespace std;
class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        return lower_bound(nums.begin(),nums.end(),target)-nums.begin();
    }
};

手动实现二分查找,注意这里要求是<=,最终返回的是left,如果是<的话,最终指向的可能是比其小的位置,也可能是比起大的位置。
python代码:

import bisect
class Solution(object):
    def searchInsert(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: int
        """
        left,right=0,len(nums)-1
        while left<=right:
            mid=left+(right-left)/2
            if nums[mid]==target:
                return mid
            else:
                if nums[mid]<target:
                    left=mid+1
                else:
                    right=mid-1
        return left

c++代码:

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

总结:
反正二分查找有很多细节需要注意,比如说这里需要注意的是使用<=和返回left,注意,经典的二分查找问题也是<=

猜你喜欢

转载自blog.csdn.net/qq_21275321/article/details/83507002