数组与矩阵 - 错误的集合

1.题目

LeetCode: 645. 错误的集合

【easy】

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

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

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

示例 1：

输入：nums = [1,2,2,4]

输出：[2,3]

示例 2：

输入：nums = [1,1]

输出：[1,2]

提示：

• $2<=nums.length<=10^4$
• $1<=nums[i]<=10^4$

2.解题

*方法一：交换元素

先通过交换数组元素，使得数组上的元素在正确的位置上，然后再遍历查找哪个元素丢失了。

java:

class Solution {
    public int[] findErrorNums(int[] nums) {
        for (int i = 0; i < nums.length; i++) {
            while (nums[i] != i + 1 && nums[nums[i] - 1] != nums[i]) {
                swap(nums, i, nums[i] - 1);
            }
        }
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] != i + 1) {
                return new int[]{nums[i], i + 1};
            }
        }
        return null;
    }
    private void swap(int[] nums, int i, int j) {
        int tmp = nums[i];
        nums[i] = nums[j];
        nums[j] = tmp;
    }
}

时间复杂度：$O(n^2)$

空间复杂度：O(1)

方法二：HashMap

参考：

https://leetcode-cn.com/problems/set-mismatch/solution/cuo-wu-de-ji-he-by-leetcode/

将数组元素和元素的数量存入哈希表，然后遍历哈希表查找重复元素和缺失元素。

java：

class Solution {
    public int[] findErrorNums(int[] nums) {
        Map<Integer, Integer> map = new HashMap<>();
        int m = -1, n = 1;
        for (int num : nums) {
            map.put(num, map.getOrDefault(num, 0) + 1);
        }
        for (int i = 1; i <= nums.length; i++) {
            if (map.containsKey(i)) {
                if (map.get(i) == 2) m = i;
            } else n = i;
        }
        return new int[]{m, n};
    }
}

时间复杂度：O(n)。遍历 nums 需要时间 O(n)，在 map 中检查每个数字需要时间 O(n)。

空间复杂度：O(n)，map 最多需要存储 1 到 n 共 n 个数字。