LeetCode刷题之数组 第35题 搜索插入位置

题目描述

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

示例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

暴力解法

思路

从前往后遍历整个数组，如何遇到大于或者等于目标值的数，则该数的索引即为目标值的索引； 如果目标值大于数组中的所有值或者数组为空，则返回值为数组的length

Java代码

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

复杂度分析

时间复杂度：O(n)
空间复杂度：O(1)

二分法

这道题目的前提是数组是有序数组，这也是使用二分查找的基础条件。以后只要看到面试题里给出的数组是有序数组，都可以想一想是否可以使用二分法。

Java代码

class Solution {
    public int searchInsert(int[] nums, int target) {
        int length = nums.length;
        int left = 0, right = length-1;                      // 循环不变量，左闭右闭
        // int middle = left + (right - left) / 2;           // 防止溢出，相当于(left+right)/2
        while (left <= right) {
            int middle = left + ((right - left) >> 1);       // >>1 代表二进制表达右移一位，相当于除以2; <<1 代表二进制表达左移一位，相当于乘以2，这样写是因为运算速度更快
            if (nums[middle] > target) {
                right = middle - 1;
            } else if (nums[middle] < target) {
                left = middle + 1;
            } else {                                         // nums[middle] = target
                return middle;
            }
        }
        return right + 1;
    }   
}

复杂度分析

时间复杂度：O(logn), 其中 n 为数组的长度。二分查找所需的时间复杂度为 O(logn)
空间复杂度：O(1)。我们只需要常数空间存放若干变量。