题目:统计一个数字在排序数组中出现的次数、及位置。
例子:
输入: nums = [5,7,7,8,8,10], target = 8
输出:2,[3,4]
输入: nums = [5,7,7,8,8,10], target = 6
输出:0,[]
方法1:暴力方法是for。
方法2:二分。
自己写的代码:
class Solution:
def func(self , nums,target):
if len(nums) == 0 or nums[0] > target or nums[-1] <target:
return 0 , {}
res = {}
n = len(nums)
i , j = 0 , n-1
while i < j:
half = (i + j ) // 2
if nums[half] < target:
i = half
elif nums[half] > target:
j = half
else:
if nums[half] not in res.keys():
res[nums[half]] = [half]
else:
if half not in res[nums[half]]:
res[nums[half]].append(half)
i += 1
return res
a = [5,7,7,8,8,10]
target = 8
s = Solution()
print(s.func(a,target))
返回一个dict,key是找到的target,value是对应的位置。