本系列博客记录学习图解算法数据结构(链接:https://leetcode.cn/leetbook/read/illustration-of-algorithm/px2ds5/)
1.剑指 Offer 03. 数组中重复的数字
找出数组中重复的数字。
在一个长度为 n 的数组 nums 里的所有数字都在 0~n-1 的范围内。数组中某些数字是重复的,但不知道有几个数字重复了,也不知道每个数字重复了几次。请找出数组中任意一个重复的数字。
我的第一想法是建立数组里每个数字的字典,出现过即赋值为True,一旦重复即return:
class Solution:
def findRepeatNumber(self, nums: List[int]) -> int:
dic = {}
for i in nums:
dic[i] = False
for n in nums:
if dic[n]:
return n
dic[n] = True
效率还可以,但占用空间较大。
翻看了一下讨论,发现我还真是忘记了好多python的基础内容,这里明显更应该用set:
class Solution:
def findRepeatNumber(self, nums: List[int]) -> int:
s = set()
for n in nums:
if n in s:
return n
s.add(n)
2.剑指 Offer 53 - I. 在排序数组中查找数字 I
统计一个数字在排序数组中出现的次数。
我知道要用二分法,可实在是不争气,在具体实现上想了半天也没有做到,最终还是参考了答案:
用二分法,1.查找右边界,2.如果没有target直接返回,3.查找左边界
class Solution:
def search(self, nums: List[int], target: int) -> int:
if not nums:
return 0
#查找右边界
i,j = 0, len(nums)-1
while i <= j:
m = (i+j)//2
if nums[m] <= target:
i = m+1
else:
j = m-1
right = i
#判断是否有target
if nums[i-1]!= target: return 0
#查找左边界
i = 0
while i <= j:
m = (i+j)//2
if nums[m] >= target:
j = m-1
else:
i = m+1
left = j
return right-left-1
利用顺序数组的特性,也可以用一个找右边界的函数来找target和target-1的右边界。
3.剑指 Offer 53 - II. 0~n-1 中缺失的数字
一个长度为n-1的递增排序数组中的所有数字都是唯一的,并且每个数字都在范围0~n-1之内。在范围0~n-1内的n个数字中有且只有一个数字不在该数组中,请找出这个数字。
有了上道题的经验,这道题就很简单了,缺少的那个数字之前,数组值与下标完全对应,那个数字之后都错开一位。所以只要找到从哪里开始错位就行了。
跟第2题类似,采用2分法就能很快找到了:
class Solution:
def missingNumber(self, nums: List[int]) -> int:
i,j = 0,len(nums)-1
while i <= j:
m = (i+j)//2
if nums[m] == m: i = m+1
else: j = m-1
return j+1