LeetCode 645. 错误的集合

目录结构

1.题目

2.题解

2.1排序

2.2辅助数组


1.题目

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

给定一个数组 nums 代表了集合 S 发生错误后的结果。你的任务是首先寻找到重复出现的整数,再找到丢失的整数,将它们以数组的形式返回。

示例:

输入: nums = [1,2,2,4]
输出: [2,3]

注意:

  1. 给定数组的长度范围是 [2, 10000]。
  2. 给定的数组是无序的。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/set-mismatch
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2.题解

2.1排序

注意有可能出现在首尾位置。

public class Solution645 {

    @Test
    public void test645() {
        int[] nums = {1, 2, 2, 4};
        System.out.println(Arrays.toString(findErrorNums(nums)));
    }

    public int[] findErrorNums(int[] nums) {
        Arrays.sort(nums);
        int dup = 0, missing = 1;
        for (int i = 0; i < nums.length - 1; i++) {
            if (nums[i] == nums[i + 1]) {
                dup = nums[i];
            } else if (nums[i + 1] - nums[i] == 2) {
                missing = nums[i] + 1;
            }
        }
        return new int[]{dup, nums[nums.length - 1] != nums.length ? nums.length : missing};
    }
}
  • 时间复杂度:O(nlogn)
  • 空间复杂度:O(1)

2.2辅助数组

public class Solution645 {

    @Test
    public void test645() {
        int[] nums = {1, 2, 2, 4};
        System.out.println(Arrays.toString(findErrorNums(nums)));
    }

    public int[] findErrorNums(int[] nums) {
        int dup = 0, missing = 0;
        int[] count = new int[nums.length + 1];
        for (int num : nums) {
            count[num]++;
        }
        for (int i = 1; i < count.length; i++) {
            if (count[i] == 0) {
                missing = i;
            } else if (count[i] == 2) {
                dup = i;
            }
        }
        return new int[]{dup, missing};
    }
}
  • 时间复杂度:O(n)
  • 空间复杂度:O(n)

你可能感兴趣的:(LeetCode)