Leetcode | 0035.搜索插入位置

题目

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

示例 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、从开头开始遍历数组，如果遇到数组中的元素和所要求的元素target相等，则返回数组中元素的下标。
2、由于数组是按照升序排列的，当在遍历的过程中遇到比target元素大的元素i时，则说明target的元素应当放在元素i前面的位置，放置成功后target元素的位置正好是元素i原来的位置。
3、当遍历到最后仍然没有得出结果时，说明target元素应当放在整个数组的末尾。

代码实现

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        if(nums[0]>target) return 0;
        int i;
        for(i = 0;i < nums.size();i ++){
            if(nums[i]==target) 
            return i;
            if(nums[i]>target)
            return i;
        }
        return i;
    }
};