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),空间复杂度O(1)

数字出现的范围是1到n是关键信息

我们也可以给 \textit{nums}[i]nums[i] 加上「负号」表示数 i+1i+1 已经出现过一次。具体地,我们首先对数组进行一次遍历。当遍历到位置 ii 时,我们考虑 \textit{nums}[\textit{nums}[i] - 1]nums[nums[i]−1] 的正负性:

如果 \textit{nums}[\textit{nums}[i] - 1]nums[nums[i]−1] 是正数,说明 \textit{nums}[i]nums[i] 还没有出现过,我们将 \textit{nums}[\textit{nums}[i] - 1]nums[nums[i]−1] 加上负号;

如果 \textit{nums}[\textit{nums}[i] - 1]nums[nums[i]−1] 是负数,说明 \textit{nums}[i]nums[i] 已经出现过一次,我们将 \textit{nums}[i]nums[i] 放入答案。

代码:

func findDuplicates(nums []int) []int {
    var res []int
    for _,v := range nums{
        if v <0{
            v = -v
        }
        if nums[v-1]<0{
            res = append(res, v)
        }else{
            nums[v-1] = -nums[v-1]
        }
        
    }
    return res
}

代码效率:

执行用时:40 ms, 在所有 Go 提交中击败了87.72%的用户
内存消耗:7.5 MB, 在所有 Go 提交中击败了63.16%的用户