剑指 Offer 03. 数组中重复的数字
题目
找出数组中重复的数字。
在一个长度为 n 的数组 nums 里的所有数字都在 0~n-1 的范围内。数组中某些数字是重复的,但不知道有几个数字重复了,也不知道每个数字重复了几次。请找出数组中任意一个重复的数字。
示例 1:
输入:
[2, 3, 1, 0, 2, 5, 3]
输出:2 或 3
思路
- hashtable。初始化dict为-1,遍历nums,碰到-1放入;反之return
代码
# hashtable
class Solution:
def findRepeatNumber(self, nums: List[int]) -> int:
hashtable = dict.fromkeys(range(len(nums)), -1)
for i in range(len(nums)):
if hashtable[nums[i]] == -1:
hashtable[nums[i]] = nums[i]
else:
return nums[i]
但是没有充分利用0~n-1的条件,改进:
class Solution:
def findRepeatNumber(self, nums: List[int]) -> int:
for i in range(len(nums)):
# 数字已在对应索引位置
if nums[nums[i]] == i: continue
# 数字对应索引已经存在相同的数,那么这个数字就是重复的
if nums[nums[i]] == nums[i]: return nums[i]
# 这里数组交换要注意前后顺序,如果先改变了nums[i],那么nums[nums[i]]的索引就会发生变化,交换失败。
nums[nums[i]], nums[i] = nums[i], nums[nums[i]]