面试题 17.19. 消失的两个数字-----leetcode

给定一个数组，包含从 1 到 N 所有的整数，但其中缺了两个数字。你能在 O(N) 时间内只用 O(1) 的空间找到它们吗？

以任意顺序返回这两个数字均可。

示例 1:

输入: [1]
输出: [2,3]
示例 2:

输入: [2,3]
输出: [1,4]
提示：

nums.length <= 30000

AC代码:

class Solution {
    public int[] missingTwo(int[] nums) {
        Arrays.sort(nums);
        int flag=1,j=0;
        int[] vis=new int[2];
        for(int i=0;i<nums.length&&j<2;i++){
            if(nums[i]!=flag){
                vis[j++]=flag;
                if(flag==nums[i]-1){
                    flag=nums[i];
                }
            }
            flag++;
        }
        while(j<2){
            if(flag!=nums[nums.length-1]){
                vis[j++]=flag++;
            }
            else{
                flag++;
            }
        }
        return vis;
    }
}