LeetCode.645—错误的集合

题目描述

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

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

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

题目分析

刚碰到这个题目想的是求集合中所有元素的和，然后与正确数组作差，再循环遍历一次，找到重复元素和位置，与差求和就是正确元素，但是这个方法行不通。然后就想到了用一个新的数组进行标记，所有出现的元素进行加一，这样重复的元素和就是2，没有出现的元素和就是1，再加上给的数组包含从1到n，所以不用担心两个0的问题。

class Solution {
    public int[] findErrorNums(int[] nums) {
    	//定义新的数组标记原来的数组
        int[] count = new int[nums.length+1];
        //遍历原来的数组，每个元素都加一
        for(int v:nums){
            count[v]++;
        }
        //存储结果
        int[] res = new int[2];
        //判断数组元素出现的次数
        for(int i = 1;i< count.length;i++){
            //结果是2就代表重复
            if(count[i]==2){
                res[0] = i;
            }else if(count[i]==0){//结果是0就代表丢失
                res[1]= i;
            }
        }
        return res;
    }
}