给定一个 n
个元素有序的(升序)整型数组 nums
和一个目标值 target
,写一个函数搜索 nums
中的 target
,如果目标值存在返回下标,否则返回 -1
。
示例 1:
输入:nums
= [-1,0,3,5,9,12],target
= 9 输出: 4 解释: 9 出现在nums
中并且下标为 4
示例 2:
输入:nums
= [-1,0,3,5,9,12],target
= 2 输出: -1 解释: 2 不存在nums
中因此返回 -1
提示:
- 你可以假设
nums
中的所有元素是不重复的。 n
将在[1, 10000]
之间。nums
的每个元素都将在[-9999, 9999]
之间。
这道题目我最开始的代码是这样的:
package ErFenChaZhao;
class Solution
{
public int search(int[] nums, int target)
{
int n = nums.length;
int low = 0;
int high = n-1;
while(low<high)
{
int middle = (low + high)/2;
if(nums[middle]<target)
{
low = middle+1;
}
else if(nums[middle]>target)
{
high = middle-1;
}
else
{
return middle;
}
}
return -1;
}
}
public class test1
{
public static void main(String[] args)
{
Solution solution = new Solution();
int[] nums = {2, 7, 11, 15}; // 示例数组
int target = 11; // 示例目标和
int num = solution.search(nums, target);
System.out.println(num); // 打印结果
}
}
运行没有问题,可在提交的时候出现了错误,发现我这个代码其实是有缺陷的,如果数组本身只有一个数,那么根本不能进入while循环,直接输出-1了,所以只需要对while的条件进行修改即可。
同时还可能考虑对代码进行一些优化,比如如果数组长度过长,那么可能会产生整数溢出,以及target根本不在数组中,不需要进入数组寻找以减少不必要的循环。
class Solution
{
public int search(int[] nums, int target)
{
// 避免当 target 小于nums[0] nums[nums.length - 1]时多次循环运算
if (target < nums[0] || target > nums[nums.length - 1])
{
return -1;
}
int n = nums.length;
int low = 0;
int high = n-1;
while(low<=high)
{
int middle = low+((high-low)>>1);
if(nums[middle]<target)
{
low = middle+1;
}
else if(nums[middle]>target)
{
high = middle-1;
}
else
{
return middle;
}
}
return -1;
}
}
public class test1
{
public static void main(String[] args)
{
Solution solution = new Solution();
int[] nums = {2}; // 示例数组
int target = 2; // 示例目标和
int num = solution.search(nums, target);
System.out.println(num); // 打印结果
}
}
部分代码参考来自代码随想录。