題目描述
給定一個排序數組和一個目標值,在數組中找到目標值,並返回其索引。如果目標值不存在於數組中,返回它將會被按順序插入的位置。
你可以假設數組中無重複元素。
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
for (int i = 0; i < nums.size() - 1; i++) {
if (nums[i] < target && nums[i+1] > target) {
return ++i;
} else if (nums[i] == target) {
return i;
} else if (nums[i] > target) {
return 0;
}
}
if (nums[nums.size()-1] >= target) {
return nums.size() - 1;
} else {
return nums.size();
}
return nums.size();
}
};