LeetCode35.搜索插入位置C++完整代码

一、题目描述

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

请必须使用时间复杂度为 O(log n) 的算法。

二、示例

1. 输入: nums = [1,3,5,6], target = 5
   输出: 2
2. 输入: nums = [1,3,5,6], target = 2
   输出: 1
3. 输入: nums = [1,3,5,6], target = 7
   输出: 4

三、具体代码

算法思想采用二分查找。

#include <iostream>
#include <vector>
using namespace std;

class Solution {
public:
    int binarySearch(int left, int right, vector<int>nums, int target) {
        int i = left;
        int j = right;
        while (i <= j) {
            int mid = i + (j - i) / 2;
            if (target > nums[mid])i = mid + 1;
            if (target < nums[mid])j = mid - 1;
            if (target == nums[mid])return mid; //如果找到了直接返回mid
        }
        //如果没找到，说明i>j,返回i或者j+1.
        return i;  //return j+1;
    }

    int searchInsert(vector<int>& nums, int target) {
        return binarySearch(0, nums.size() - 1, nums, target);
    }
};

int main() {
    vector<int>nums = {1,3,5,6}; //示例2
    Solution S;
    cout << S.searchInsert(nums,2) << endl;
	system("pause");
	return 0;
}

时间复杂度为O(log n),空间复杂度为O(1)。