今天做的是LeetCode35,寻找给定元素在数组中的位置,如果没有此元素的话给出插入的位置
基本解法很简单:
public int searchInsert(int[] nums, int target) {
for(int i=0;i<nums.length;i++){
if(nums[i]==target||nums[i]>target)
return i;
}
return nums.length;
}
又想了一下,
最基础的方法就是从头开始向后搜索
如果遇见等于的就直接返回序号
如果第一次遇见大于的就返回当前序号--
如果想提高点效率的话可以采用二分搜索
从中间开始
如果等于中间值的话直接返回
如果小于的话在左边寻找
如果大于的话在右部寻找
很显然要利用到递归算法,代码如下:
public int searchInsert(int[] nums, int target) {
return searchBinaryInsert(nums, target);
}
public int searchBinaryInsert(int[] nums, int target) {
if(nums.length<=2){
for(int i=0;i<nums.length;i++){
if(target==nums[i]||target<nums[i])
return i;
}
return nums.length;
}
int mediaValue=nums[nums.length/2];
if(target==mediaValue)
return nums.length/2;
else if(target<mediaValue){
int[] left=new int[nums.length/2];
System.arraycopy(nums, 0, left, 0, nums.length/2);
return searchBinaryInsert(left, target);
}else{
int[] right=new int[(nums.length-1)/2];
System.arraycopy(nums, nums.length/2+1, right, 0, (nums.length-1)/2);
return searchBinaryInsert(right, target)+nums.length/2+1;
}
}
重点是在右边寻找的时候别忘加了左边数组的长度
查找了一下大神的二分法,发现根本不用递归,只需要一个while循环块就做到了,代码如下:
public int searchInsert(int[] A, int target) {
int low = 0, high = A.length-1;
while(low<=high){
int mid = (low+high)/2;
if(A[mid] == target) return mid;
else if(A[mid] > target) high = mid-1;
else low = mid+1;
}
return low;
}
我以为这种优质解法应该没几个人想到,我之前自己用的方法就算不错了。
结果前几种都是用这个方法写的
尴尬啊
不过知道二分查找真的不需要用递归了!!!!!!!!!