LeetCode-35-搜索插入位置

版权声明：禁止盗用 https://blog.csdn.net/weixin_43411988/article/details/89454726

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

你可以假设数组中无重复元素。

• 示例 1:

输入: [1,3,5,6], 5
输出: 2

• 示例 2:

输入: [1,3,5,6], 2
输出: 1

• 示例 3:

输入: [1,3,5,6], 7
输出: 4

• 示例 4:

输入: [1,3,5,6], 0
输出: 0

1. Python

import bisect

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

2. CPP:

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