来源:力扣(LeetCode)给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。假设数组中无重复元素。
示例 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:根据题意,这是一个有序数组,正常思路用二分法找数组中它的位置,如果存在则一定找到,不存在,则就在你最后找的left或right位置,这里就需要你判断一哈就可以了。直接上代码(我用C++写的):
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
if(nums.size()==0)
return 0;
int left=0;
int right=nums.size()-1;
int mid=0;
while(left<right)
{
mid=(left+right)>>1;
if(nums[mid]==target)
return mid;
else if(nums[mid]>target)
right=mid-1;
else
left=mid+1;
}
if(target>nums[left])
left+=1;
return left;
}
};
第二种思路,因为是有序数组,所以直接在数组中查找,找第一个不小于目标值的数,则该下标就是目标的位置(如果目标值不存在,则该位置就是他的插入位置),如果,目标值不在数组中且比数组中所有数都大,则插入在最后。直接上代码:
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
for(int i=0;i<nums.size();i++)
{
if(nums[i]>=target)//因为是排序的,第一个大于他的就是他的位置
return i;
}
//都比他小
return nums.size();
}
};
如果有错,欢迎指正~