题目描述:
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-insert-position
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路:
简单粗暴来看这道题,本身是有序数组,二分法查找十分方便,如果二分法能够找到,就直接返回找到数位置。如果找不到,就遍历数组找到第一个比这个数小的数,直接返回这个数的位置(插进去了,这个数就占了原来的这个位置)。
注意:数组的大小一旦初始化确认之后就无法改变了,所以无法做到把这个数再加到数组里面。如果要做这一步,必须要创建一个新的数组。
代码(Java):
public class doingmyself {
public static void main(String[] args) {
int[] nums = new int[] {1,3,5,6};
int target = 7;
System.out.println(searchInsert(nums, target));
}
public static int searchInsert(int[] nums, int target) {
int ans = binarySearch(nums, target); //二分法查找,如果找得到就返回位置,找不到返回-1
boolean flag = true; //这个标志位表示数组所有的数都没有插入进来的数大,需要把这个数挂到数组后面
if(ans==-1) { //找不到的时候,只要判断这个数插进哪个位置
for(int i=0;i<nums.length;i++) { //遍历数组,找到数组里面第一个比这个数大的数的位置
if(target<nums[i]) {
ans = i;
flag = false; //找到了这个数,就不需要在数组后面挂这个数。
break;
}
}
if(flag) ans = nums.length;
}
return ans;
}
public static int binarySearch(int[] nums, int target) { //二分法查找
int ans = -1;
int low = 0;
int high = nums.length-1;
int middle = 0;
while(high>=low) {
middle = low + (high - low)/2;
if(nums[middle]==target) {
ans = middle;
}
if(nums[middle]>=target) {
high = middle - 1;
}else if (nums[middle]<=target) {
low = middle + 1;
}
}
return ans;
}
}