一道有意思的题目
442. 数组中重复的数据
给你一个长度为 n
的整数数组 nums
,其中 nums
的所有整数都在范围 [1, n]
内,且每个整数出现 一次 或 两次 。请你找出所有出现 两次 的整数,并以数组形式返回。
你必须设计并实现一个时间复杂度为 O(n)
且仅使用常量额外空间(不包括存储输出所需的空间)的算法解决此问题。
示例 1:
输入:nums = [4,3,2,7,8,2,3,1] 输出:[2,3]
示例 2:
输入:nums = [1,1,2] 输出:[1]
示例 3:
输入:nums = [1] 输出:[]
提示:
n == nums.length
1 <= n <= 105
1 <= nums[i] <= n
nums
中的每个元素出现 一次 或 两次
思路:
题目要求实现一个时间复杂度为 O(n)
且仅使用常量额外空间,可以考虑就在原数组上面改动。
但是改数组的第一个值是,就可能把其他地方的值给改变了。如果我们用取模运算的话就可以实现对后面数组中数字的余数不改变,但是数值会改变。对于
nums = [4,3,2,7,8,2,3,1]
对于第一个元素4,---> nums[4%n-1]=num[4%n-1]+n
为什么要减一?要保证数组的下标可以一一对应起来。
为什么要%n,这样下一次对原数组操作时,取余就可以得到原来的值
如果出现两次4,数组就会加两次4.最后如果数组中有>2n的值出现是,就说明该数字出现了两次
代码如下:
class Solution(object):
def findDuplicates(self, nums):
"""
:type nums: List[int]
:rtype: List[int]
"""
ans=[]
n=len(nums)
for i in range(n):
nums[(nums[i])%n-1]+=n
for i in range(n):
if nums[i]>2*n:
ans.append(i+1)
return ans
给你一个含 n
个整数的数组 nums
,其中 nums[i]
在区间 [1, n]
内。请你找出所有在 [1, n]
范围内但没有出现在 nums
中的数字,并以数组的形式返回结果。
示例 1:
输入:nums = [4,3,2,7,8,2,3,1] 输出:[5,6]
示例 2:
输入:nums = [1,1] 输出:[2]
提示:
n == nums.length
1 <= n <= 105
1 <= nums[i] <= n
进阶:你能在不使用额外空间且时间复杂度为 O(n)
的情况下解决这个问题吗? 你可以假定返回的数组不算在额外空间内。
解题思维:
与上文类似
参考答案如下:
class Solution(object):
def findDisappearedNumbers(self, nums):
"""
:type nums: List[int]
:rtype: List[int]
"""
ans=[]
n=len(nums)
for i in range(0,n):
nums[nums[i]%n-1]+=n
print(nums)
for i in range(0,n):
if nums[i]<=n:
ans.append(i+1)
return ans