Find All Duplicates in an Array 题解

442. Find All Duplicates in an Array

题目描述：

Given an array of integers, 1 ≤ a[i] ≤ n (n = size of array), some elements appear twice and others appear once.

Find all the elements that appear twice in this array.

Could you do it without extra space and in O(n) runtime?

Example:

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

Output:
[2,3]

题目链接：442. Find All Duplicates in an Array

算法描述：

根据题意，我们需要在给定的一个序列中找到具有重复的数，结果将返回一个只带有重复数的数组。按照题目要求，我们要控制空间复杂度在 O（1），时间复杂度控制在 O（n）。

细看题目中数组的条件：1 ≤ a[i] ≤ n (n = 数组的长度)，那么首先，我们对该数组 nums 进行初始遍历，根据它们的值将它们安置在指定的位置，如数组中的1，会把它放置在首元素的位置 nums[0]；数组中的2，会把它放置在第二元素的位置 nums[1]；数组中的4，会把它放置在第四元素的位置 nums[3]（注意：在完成交换之后要依然继续判断新的元素是否满足该条件）……遍历之后即可得到初始序列。

第二步，从头到尾遍历初始序列，如果发现它们的值与它们所处的位置不符，即为重复元素，将它们加入结果数组ans[]。算法时间复杂度为 O（n），空间复杂度为 O（1），代码如下：

代码：

class Solution {
public:
    vector<int> findDuplicates(vector<int>& nums) {
        vector<int> ans;
        for(int i=0; i<nums.size(); i++){
            if( nums[i] != nums[nums[i]-1] ){  
                swap(nums[i], nums[nums[i]-1]);
                i--;
            }
        }
        
        for(int i=0; i<nums.size(); i++){
            if( nums[i]!=i+1 ){
                ans.push_back(nums[i]);
            }
        }
        
        return ans;
    }
};