【LeetCode】645.错误的集合

题目

集合 s 包含从 1 到 n 的整数。不幸的是，因为数据错误，导致集合里面某一个数字复制了成了集合里面的另外一个数字的值，导致集合 丢失了一个数字 并且 有一个数字重复 。

给定一个数组 nums 代表了集合 S 发生错误后的结果。

请你找出重复出现的整数，再找到丢失的整数，将它们以数组的形式返回。

思路

利用map统计出现的次数。出现次数为0，则是丢失的整数；出现次数为2，则是重复的整数。

func findErrorNums(nums []int) []int {
    // cnt := map[int]int{}
    cnt := make(map[int]int)
    for _, v := range nums {
        cnt[v]++
    }
    ans := make([]int, 2)
    for i := 1; i <= len(nums); i++ {
        c := cnt[i]
        if c == 2 {
            ans[0] = i
        } else if c == 0 {
            ans[1] = i
        }
    }
    return ans
}