645.错误的集合Java

645.错误的集合Java

题目描述

集合 s 包含从 1 到 n 的整数。集合里面某一个数字复制了成了集合里面的另外一个数字的值，导致集合 丢失了一个数字并且有一个数字重复 。
给定一个数组 nums 找出重复出现的整数，再找到丢失的整数,以数组的形式返回。

输入输出样式

示例1：
输入：nums = [1,2,2,4]
输出：[2,3]

示例2：
输入：nums = [1,1]
输出：[1,2]

本题来自LeetCode：https://leetcode-cn.com/problems/set-mismatch/

思路

sum(nums) - sum(set(nums)) = 重复的数字
(1 + len(nums)) * len(nums) / 2 - sum(set(nums)) = 丢失的数字

算法分析

时间复杂度O(n），空间复杂度为O(n)

求解函数

public int[] findErrorNums(int[] nums) {
        Set<Integer> set = new HashSet();
        int[] rst = new int[2];
        int sum = 0;
        for (int item : nums) {
            if (!set.add(item)) {
                rst[0] = item;
                continue;
            }
            sum += item;
        }
        rst[1] = nums.length * (nums.length + 1) / 2 - sum;
        return rst;
    }