public int[] searchRange(int[] A, int target) {
int[] result = {-1, -1};
if (A == null || A.length == 0) {
return result;
}
int left = 0;
int right = A.length - 1;
int mid;
while (left + 1 < right) {
mid = left + (right - left) / 2;
if (A[mid] == target) {
// 找到target后不停止,而是继续向左找,因为要找到第一个target出现的地方
// 因为这个template的写法不是 mid - 1, 所以即使继续向左找,已经找到的target也不会丢失
right = mid;
} else if (A[mid] < target) {
left = mid;
} else {
right = mid;
}
}
// 因为要找第一个出现的target,
// 所以优先比较left
if (A[left] == target) {
result[0] = left;
} else if (A[right] == target) {
result[0] = right;
}
left = 0;
right = A.length - 1;
while (left + 1 < right) {
mid = left + (right - left) / 2;
if (A[mid] == target) {
// 找到target后不停止,而是继续向右找,因为要找到最后一个target出现的地方
left = mid;
} else if (A[mid] < target) {
left = mid;
} else {
right = mid;
}
}
// 因为要找到最后一个target出现的地方,因此优先比较right
if (A[right] == target) {
result[0] = left;
} else if (A[left] == target) {
result[0] = right;
}
return result;
}
大题的思路是使用两次 binary search 的模板,
第一次找到target第一次出现的位置(最左边),
第二次找到target最后一次出现的位置(最右边)。
方法就是,当找到target之后,不像常规方法一次退出,而是继续搜索左半边(或右半边)。