给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 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
思路:由于是有序数组,并且要使用O(log n)的算法,那么就只能使用二分法了,for循环的话是O(n);如果刚好是数组中存在的,那么二分法直接返回即可,但是如果二分法没有找到的话,可能就需要返回low值
package com.dlh.test.算法;
//给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
//请必须使用时间复杂度为 O(log n) 的算法。
import java.util.Scanner;
public class 搜索插入位置 {
public static void main(String[] args) {
Scanner sc =new Scanner(System.in);
String[] string = sc.nextLine().split(" ");
int target = sc.nextInt();
int[] nums = new int[string.length];
for (int i = 0; i < string.length; i++) {
nums[i] = Integer.parseInt(string[i]);
}
int result = searchInsert(nums,target);
System.out.println(result);
}
private static int searchInsert(int[] nums, int target) {
//由于是有序数组,并且要使用O(log n)的算法,那么就只能使用二分法了,for循环的话是O(n);
int low = 0;
int high = nums.length -1;
int result = 0;
while (low <= high){
int middle = (low+high)/2;//寻找二分法中间值
if (target == nums[middle]){//如果二分法找到对应的值,直接返回
result = middle;
return result;
}else if (target < nums[middle]){//没找到的话就依次替换high和low的值
high = middle-1;
}else {
low = middle+1;
}
}
//如果走到这一步,说明数组中没有对应的值,那么返回low值即可
return low;
}
}