Leetcode 35.搜索插入位置

题目描述

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

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

示例 1:

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

示例 2:

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

示例 3:

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

示例 4:

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

题目分析

由于是有序数组, 则先比较目标数字与两侧极值的, 确定目标数字是否在数组之中, 而后从第一个元素开始迭代, 如果target > nums[i] 并且 <= nums[i+1]则返回i+1;

源码

1
2
3
4
5
6
7
8
9
10
11
12
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
if(target <= nums[0]) return 0;
if(target > nums[nums.size()-1]) return nums.size();
if(target == nums[nums.size()-1]) return nums.size()-1;
for(int i=0; i<nums.size(); ++i) {
if(nums[i] < target && nums[i+1] >= target) return i+1;
}
return -1;
}
};
0%