442. 数组中重复的数据 中等

给定一个整数数组 a,其中1 ≤ a[i] ≤ n (n为数组长度), 其中有些元素出现两次而其他元素出现一次。

找到所有出现两次的元素。

你可以不用到任何额外空间并在O(n)时间复杂度内解决这个问题吗?

示例:

输入:
[4,3,2,7,8,2,3,1]

输出:
[2,3]

代码参考:

package main

import "fmt"

func main() {
    fmt.Println(findDuplicates([]int{4, 3, 2, 7, 8, 2, 3, 1})) // [2, 3]
}

// 多么适合用哈希表的题...
// 可惜不能使用额外空间,复杂度 O(N),可能要求在数组原地操作、一次遍历解决
// 解法和下方的 448 很像,充分利用 1~n 的已知条件,十分巧妙
func findDuplicates(nums []int) []int {
    var dups []int
    for _, num := range nums {
        if num < 0 {
            num = -num
        }

        n := num - 1
        if nums[n] < 0 { // 指向的位置以为负值,则重复
            dups = append(dups, num)
            continue
        }
        nums[n] = -nums[n]
    }
    return dups
}
最后编辑: kuteng  文档更新时间: 2021-06-05 10:16   作者:kuteng