645. 错误的集合

645. 错误的集合

代码：

class Solution {
    public int[] findErrorNums(int[] nums) {
        int[] error = new int[2];
        int[] count = new int[nums.length + 1];
        count[0] = -1;
        for (int i = 0; i < nums.length; i++) count[nums[i]]++;
        for (int i = 1; i <= nums.length; i++)
        {
            if (count[i] == 0) error[1] = i;
            else if (count[i] > 1) error[0] = i;
        }
        return error;
    }
}