今天做的是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;
}
我以爲這種優質解法應該沒幾個人想到,我之前自己用的方法就算不錯了。
結果前幾種都是用這個方法寫的
尷尬啊
不過知道二分查找真的不需要用遞歸了!!!!!!!!!