LeetCode-35-搜索插入位置

最新推荐文章于 2024-03-06 08:30:00 发布

fffzlfk

最新推荐文章于 2024-03-06 08:30:00 发布

阅读量105

点赞数

分类专栏： Python 文章标签： LeetCode

本文链接：https://blog.csdn.net/weixin_43411988/article/details/89454726

版权

Python 专栏收录该内容

19 篇文章 1 订阅

订阅专栏

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

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

示例 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

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))

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;
    }
};