leetcode 448. Find All Numbers Disappeared in an Array

目录

一、问题描述

二、代码实现

思路一

思路二

---

 

https://leetcode.com/problems/find-all-numbers-disappeared-in-an-array/

给定一个元素值在[1, n]范围的长度为n的整数数组，其元素有的出现两次有的出现一次，找到[1, n]区间中所有没有在数组中出现的元素。要求时间O(n)，空间O(1)，空间不包括返回的列表。

 

一、问题描述

测试用例：

Example:

Input:
[4,3,2,7,8,2,3,1]

Output:
[5,6]

思路一：由于元素值范围是[1, n]，数组下标范围是[0, n-1]，所以我们可以遍历一遍，将元素nums[i]存放到下标为nums[i]-1的位置处。然后再遍历一遍，如果nums[i] != i+1，则说明区间值i+1没有出现在数组中。

思路二：与思路一类似，不过不移动元素的位置，而是对于元素nums[i]，将下标为nums[i]-1的元素变为负数或添加上某个标志（如将其值增加到大于n）。当第二次遍历时，如果有负数或其他标志，说明下标i所代表的数值i+1在数组中存在，反之则不存在，添加到结果列表中。

 

二、代码实现

思路一

class Solution {

    public List<Integer> findDisappearedNumbers(int[] nums) {
        List<Integer> result = new ArrayList<>();
        if (nums == null || nums.length == 0) {
            return result;
        }
        
        //first pass
        for (int i=0; i<nums.length; ) {
            if ((nums[i] != i+1) && (nums[nums[i]-1] != nums[i])) {
                swap(nums, i, nums[i]-1);
            } else {
                i++;
            }
        }
        //System.out.println(Arrays.toString(nums));
        
        //second pass
        for (int i=0; i<nums.length; i++) {
            if (nums[i] != i+1) {
                result.add(i+1);
            }
        }
        return result;
    }
    
    private void swap(int[] nums, int i, int j) {
        if (i != j) {
            nums[i] = nums[i] + nums[j];
            nums[j] = nums[i] - nums[j];
            nums[i] = nums[i] - nums[j];
        }
    }
}

思路二

class Solution {

    //思路二的第二种写法
    public List<Integer> findDisappearedNumbers(int[] nums) {
        List<Integer> res = new ArrayList<>();
        int n = nums.length;
        for (int i = 0; i < nums.length; i ++) nums[(nums[i]-1) % n] += n;  //元素出现k次，则元素加上k*n
        for (int i = 0; i < nums.length; i ++) if (nums[i] <= n) res.add(i+1);
        return res;
    }
    
    //思路一的第一种写法
    public List<Integer> findDisappearedNumbers2(int[] nums) {
        List<Integer> result = new ArrayList<>();
        if (nums == null || nums.length == 0) {
            return result;
        }
        
        //first pass
        for (int i=0; i<nums.length; i++) {
            int index = Math.abs(nums[i]) - 1;
            if (nums[index] > 0) {
                nums[index] = -1 * nums[index];
            }
        }
        
        //second pass
        for (int i=0; i<nums.length; i++) {
            if (nums[i] > 0) {
                result.add(i+1);
            }
        }
        return result;
    }
    
}

 

参考：

https://leetcode.com/problems/find-all-numbers-disappeared-in-an-array/discuss/92958/c%2B%2B-solution-O(1)-space

https://leetcode.com/problems/find-all-numbers-disappeared-in-an-array/discuss/93007/Simple-Java-In-place-sort-solution

https://leetcode.com/problems/find-all-numbers-disappeared-in-an-array/discuss/92980/5-line-Java-Easy-understanding

https://leetcode.com/problems/find-all-numbers-disappeared-in-an-array/discuss/92957/2ms-O(n)-In-Space-Java

https://leetcode.com/problems/find-all-numbers-disappeared-in-an-array/discuss/93163/O(N)-Time-%2B-O(1)-Space