在排序数组中查找数字 I
统计一个数字在排序数组中出现的次数。
- 方法一为二分法,时间复杂度O(logN),空间复杂度O(1)
- 其他方法:哈希表时间空间复杂度都为O(N)
二分法,参考:
class Solution:
def search(self, nums: List[int], target: int) -> int:
# if len(nums) == 1:
# if nums[0]!=target:
# return(0)
# return(1)
# if len(nums) == 0:
# return(0)
n = len(nums)
i, j = 0, len(nums) - 1
#先找右边的
while i <= j:#注意这里是小于等于不是小于,小于没法处理像【2,2,2】这样的
m = (i + j)//2
if nums[m] <= target:#注意右边界是小于等于,之所以小于等于是要把i移动到target的右边
i = m + 1
else:
j = m - 1
right = i
if j > 0 and nums[j] != target:#加个j>0是为了防止空列表情况
return(0)
i = 0
#再找左边的
while i <= j:
m = (i + j)//2
if nums[m] < target:#左边界是小于
i = m + 1
else:
j = m -1
left = j
return(right - left -1)