这题看似很简单,一个字典就结束了。但仔细看题的话,还有更优秀的解法。本题正好是0到n-1填满整个数组,如果排序的话只要看每个元素是否在对应的位置就好,而利用这种思路,咱们可以根据当前元素和对应位置的元素进行交换,出现重复则说明位置满了。
class Solution:
def findRepeatNumber(self, nums: List[int]) -> int:
for i, m in enumerate(nums):
while True:
if m == nums[m]:
if m != i: return m
else: break
nums[i], nums[m] = nums[m], nums[i]
问题变种:题目相同,但不能修改输入数组
书中提供了一种二分的思想,可惜这种思想是错误的。即不断二分划分区间,看在这个区间索引范围的元素是否大于区间长度。错误的地方在于,一旦两边相等,我们就无法判断哪边出了问题,需要重新找划分位置。因此,此算法的复杂度为O(N方)。
因此,在这样的条件下,牺牲时间换空间是很不值的。
class Solution:
def findRepeatNumber(self, nums: List[int]) -> int:
l, r = 0, len(nums)-1
while l < r:
mid = (l+r) // 2 # 剩两个数时, left=mid就要用ceil判定中位数
sum0 = 0
for m in nums:
if l <= m <= mid:
sum0 += 1
if sum0 > mid-l+1: r = mid
else: l = mid+1
return l