题目要求:
分析:
从题目中可以看到,给出的nums数组是有序的,并且要求时间复杂度是log n,那么肯定使用二分法。
这里要求我们找到给定元素的第一个和最后一个位置,那么首先我们需要找到指定的元素,如果没有这个元素,就返回[-1, -1],如果有这个元素,那么我们还需要进行如下操作:
- 在查找到的元素的左边进行查找,如果有与target相等的元素,则记录下它的坐标,直到找不到与target相等的元素为止;
- 在查找到的元素的右边也进行相同的操作。
我们以题目中给出的nums为例,如下图所示:
理清思路之后,我们很快就能写出代码了。
具体代码如下:
class Solution {
public int[] searchRange(int[] nums, int target) {
int left = 0, right = nums.length - 1;
int temp = -1;
while(left <= right) {
int mid = (left + right) / 2;
if(target > nums[mid]) {
left = mid + 1;
} else if(target == nums[mid]) {
temp = mid;
break;
} else {
right = mid - 1;
}
}
if(temp == -1) {
return new int[] {-1, -1};
} else {
int temp1 = temp, temp2 = temp;
while(temp1 > 0 && nums[temp1 - 1] == target) {
temp1 --;
}
while(temp2 < nums.length - 1 && nums[temp2 + 1] == target) {
temp2 ++;
}
return new int[] {temp1, temp2};
}
}
}
注意:
在这里一定要注意边界问题!!!
一定一定要注意!!!
我在调试代码的过程中,出现了以下问题:
这是因为我在循环过程中忽略了left == right的情况。
总之,二分查找虽然实现简单,但是边界也是够让人喝一壶的!