查找已由小到大排序的向量vector中target的位置并返回下标,如不存在则返回应插入的位置的下标。
我的答案:
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int size = nums.size();
int i;
if(nums.empty())
return 0;
for(i=0;i<size;i++)
if(target==nums[i]) // 存在target
return i;
for(i=0;i<size;i++) // 不存在target
if(target < nums[i])
return i;
return i; // 在最后一个
}
};
比较慢,原因是对vector遍历了。查看更快的答案,是采用二分法来寻找位置的,更快:
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int l = 0, r = nums.size()-1;
if(target > nums.back()) return nums.size();
while(l < r){
int mid = l + (r-l)/2; // 二分
if(nums[mid] == target) return mid;
else if(nums[mid] < target ) l = mid + 1;
else r = mid;
}
return r;
}
};