LeetCode-35. 搜尋插入位置
阿新 • • 發佈:2018-11-24
題目
給定一個排序陣列和一個目標值,在陣列中找到目標值,並返回其索引。如果目標值不存在於陣列中,返回它將會被按順序插入的位置。
你可以假設陣列中無重複元素。
示例 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
解題
- 最簡單的辦法, 就是遍歷一遍, 時間複雜度, 最優O(1), 最差O(n), 程式碼如下:
class Solution {
public int searchInsert(int[] nums, int target) {
for (int i = 0; i < nums.length; i++)
if (nums[i] >= target)
return i;
return nums.length;
}
}
- 因為是有序陣列, 所以可以使用二分搜尋來處理, 在資料量較大的情況下, 二分搜尋的效率更高, 時間複雜度是O(logn), LeetCode給的測試用例資料量都不大, 所以二分搜尋的效能並不高於直接遍歷的方法, 程式碼如下:
class Solution35 {
public int searchInsert(int[] nums, int target) {
int index = findIndex(nums, 0, nums.length - 1, target);
return index;
}
private int findIndex(int[] nums, int left, int right, int target){
if(target > nums[right]){
return right + 1 ;
}
if(target < nums[left]){
return left;
}
if(right == left){
return right;
}
int mid = (right - left + 1) / 2 + left;
if(target == nums[mid]){
return mid;
} else if (target > nums[mid]){
return findIndex(nums, mid + 1, right, target);
} else {
return findIndex(nums, left, mid - 1, target);
}
}