leetcode-448-找到所有数组中消失的数字-java

题目及测试

package pid448;
/*448. 找到所有数组中消失的数字

给定一个范围在  1 ≤ a[i] ≤ n ( n = 数组大小 ) 的 整型数组，数组中的元素一些出现了两次，另一些只出现一次。

找到所有在 [1, n] 范围之间没有出现在数组中的数字。

您能在不使用额外空间且时间复杂度为O(n)的情况下完成这个任务吗? 你可以假定返回的数组不算在额外空间内。

示例:

输入:
[4,3,2,7,8,2,3,1]

输出:
[5,6]




*/

import java.util.List;

public class main {
	
	public static void main(String[] args) {
		int [][] testTable = {{4,3,2,7,8,2,3,1},{1,2,3}};
		for (int[] ito : testTable) {
			test(ito);
		}
	}
		 
	private static void test(int[] ito) {
		Solution solution = new Solution();
		List<Integer> rtn;
		long begin = System.currentTimeMillis();
		for(int i=0;i<ito.length;i++){
			System.out.print(ito[i]+"  ");
		}
		System.out.println();
		//开始时打印数组
		
		rtn= solution.findDisappearedNumbers(ito);//执行程序
		long end = System.currentTimeMillis();	
		
		System.out.println("rtn=" );
		for(int i=0;i<rtn.size();i++){
			System.out.print(rtn.get(i)+" ");
		}
		System.out.println();
		System.out.println("耗时：" + (end - begin) + "ms");
		System.out.println("-------------------");
	}

}

解法1（成功，24ms，超慢）

建立一个set，里面装入1-n，然后不断remove(nums[i])，剩下的数就是result

package pid448;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

class Solution {
    public List<Integer> findDisappearedNumbers(int[] nums) {
    	int length = nums.length;
    	if(length == 0){
    		return new ArrayList<Integer>();
    	}
    	Set<Integer> set = new HashSet<>();
    	for(int i=1;i<length+1;i++){
    		set.add(i);
    	}
    	for(int i=0;i<length;i++){
    		set.remove(nums[i]);
    	}
    	List<Integer> result = new ArrayList<>();
    	for(Integer i:set){
    		result.add(i);
    	}    	
    	return result;
    }
}

解法2（成功，7ms，较快）

原地修改

我们需要知道数组中存在的数字，由于数组的元素取值范围是 [1, N]，所以我们可以不使用额外的空间去解决它。

我们可以在输入数组本身以某种方式标记已访问过的数字，然后再找到缺失的数字。

遍历输入数组的每个元素一次。

我们将把 |nums[i]|-1 索引位置的元素标记为负数。即 nums[nums[i]−1] × −1

然后遍历数组，若当前数组元素 nums[i] 为负数，说明我们在数组中存在数字 i+1。

	public List<Integer> findDisappearedNumbers(int[] nums) {
		int length = nums.length;
    	if(length == 0){
    		return new ArrayList<Integer>();
    	}
    	for(int i=0;i<length;i++) {
    		int now = Math.abs(nums[i]);
    		if(nums[now - 1] > 0) {
    			nums[now - 1] = -nums[now - 1];
    		}
    	}
    	List<Integer> result = new ArrayList<>();
    	for(int i=0;i<length;i++) {
    		int now = nums[i];
    		if(now > 0) {
    			result.add(i + 1);
    		}
    	}
    	return result;
	}