搜索插入位置(Java實現)
給定一個排序數組和一個目標值,在數組中找到目標值,並返回其索引。如果目標值不存在於數組中,返回它將會被按順序插入的位置。
你可以假設數組中無重複元素。
示例 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
我的思路很簡單,就是先遍歷查找,藉助數據結構ArrayList,如果找到就返回下標;找不到就add進ArrayList中,然後對ArrayList排序,使用indexOf(target)來返回target的下標即可。
代碼實現之:
package Day49;
import java.util.ArrayList;
import java.util.Collections;
/**
* @Author Zhongger
* @Description 給定一個排序數組和一個目標值,在數組中找到目標值,並返回其索引。如果目標值不存在於數組中,返回它將會被按順序插入的位置。
* 你可以假設數組中無重複元素。
* @Date 2020.3.23
*/
public class SearchInsertSolution {
public static void main(String[] args) {
int[] nums={1,3,5,6};
int target=7;
System.out.println(new SearchInsertSolution().searchInsert(nums, target));
}
public int searchInsert(int[] nums, int target) {
if (nums==null){
return -1;
}
if (nums.length<=0){
return -1;
}
ArrayList<Integer> list = new ArrayList<>();
for (int i = 0; i < nums.length; i++) {
list.add(nums[i]);
if (target==nums[i]){
return i;
}
}
list.add(target);
Collections.sort(list);
return list.indexOf(target);
}
}
還有一種方法就是二分查找法,簡單得很:
public int searchInsert2(int[] nums, int target) {
int left = 0, right = nums.length - 1;
while(left <= right) {
int mid = (left + right) / 2;
if(nums[mid] == target) {
return mid;
} else if(nums[mid] < target) {
left = mid + 1;
} else {
right = mid - 1;
}
}
return left;
}