LeetCode - 35 - Tung in ECNU


LeetCode - 35

LeetCode - 35

https://leetcode-cn.com/problems/search-insert-position/

题目

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

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

示例 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. 要是找到了就返回
  2. 没找到就返回要插入的位置

时间复杂度为 $$O(N)$$

暴力法

public int searchInsert(int[] nums, int target) {
    if (target < nums[0]) return 0;
    if (target > nums[nums.length - 1]) return nums.length;
    for (int i= 0; i < nums.length; i++) {
        if (target > nums[i] && target < nums[i + 1]) {
            return i + 1;
        } else if (target == nums[i]) {
            break;
        }
    }
    return 0;
}

二分法

二分法基本模板

class Solution {
    public int searchInsert(int[] nums, int target) {
        int left = 0, right = nums.length - 1; 
        while(left <= right) { 
            int mid = (left + right) / 2; 
            if(nums[mid] == target) { 
                  //code here
            } else if(nums[mid] < target) {
                left = mid + 1; 
            } else {
                right = mid - 1; 
            }
        }
        return 0;
    }
}

在暴力法的基础上根据二分法模板进行改进:

由于数组是有序的,可以把遍历查找的过程换成二分查找,时间复杂度为 $$O(log N)$$

代码如下:

class Solution {
    public int searchInsert(int[] nums, int target) {
        int left = 0, right = nums.length - 1;
        while(left <= right) {
            int mid = (left + right) / 2;
            if(nums[mid] == target) {
                return mid;
            } else if(nums[mid] < target) {
                left = mid + 1;
            } else {
                right = mid - 1;
            }
        }
        return left;
    }
}

本文链接:

https://noahtung.xyz/index.php/archives/20/