leetcode 34. 在排序数组中查找元素的第一个和最后一个位置

题解

分别进行两次二分查找,找出最左边和最右边的位置

示例代码(go)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
func searchRange(nums []int, target int) []int {
left, right := -1, -1
n := len(nums)
for i, j := 0, n-1; i <= j; {
mid := (i + j) / 2
if nums[mid] > target {
j = mid - 1
} else if nums[mid] < target {
i = mid + 1
} else {
left = mid
j = mid - 1
}
}
for i, j := 0, n-1; i <= j; {
mid := (i + j) / 2
if nums[mid] > target {
j = mid - 1
} else if nums[mid] < target {
i = mid + 1
} else {
right = mid
i = mid + 1
}
}
return []int{left, right}
}