给定一个整数数组 a,其中1 ≤ a[i] ≤ n (n为数组长度), 其中有些元素出现两次而其他元素出现一次。
找到所有出现两次的元素。
你可以不用到任何额外空间并在O(n)时间复杂度内解决这个问题吗?
示例:
输入:
[4,3,2,7,8,2,3,1]
输出:
[2,3]
解法:
1. O(N^2): 常规解题思路,先排序。再遍历,如果两个相邻元素相等,则会有重复元素。存入list。
public List<Integer> findDuplicates(int[] nums) { List<Integer> list = new ArrayList<>(); Arrays.sort(nums); for (int i=1;i<nums.length;i++) { if (nums[i]==nums[i-1]) { list.add(nums[i]); } } return list; }
2. O(N): 题目已知数组元素的大小<=数组长度。因此可以借助数组工具,如果查阅过该元素值,则把数组值置为负数,表示已阅。当如果再次查到这个负值,意味着第二次遍历到重复元素。
public List<Integer> findDuplicates(int[] nums) { List<Integer> result = new ArrayList<Integer>(); if(nums == null) return result; for(int i=0; i<nums.length; i++){ int location = Math.abs(nums[i])-1; if(nums[location] < 0){ result.add(Math.abs(nums[i])); }else{ nums[location] = -nums[location]; } } return result; }