[Leetcode] 442. Find All Duplicates in an Array 解题报告

来源:互联网 发布:贸易结构优化 编辑:程序博客网 时间:2024/05/28 23:21

题目

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]

思路

要求不能用额外空间,那么就只能inplace的修改数组了。我们的思路是:遍历数组,将value = nums[i]交换到nums[value - 1]的位置,这样在当发现两个要交换的位置上的数字相同时,则说明找到了一个重复数字。为了避免后面继续再找到,我们将其中一个置为0。

那么算法的时间复杂度怎么分析呢?虽然在for循环里面还嵌套了一个while循环,但是仔细观察可以发现,每个value最多被移动一次,这样实际上整个for循环执行下来,while被执行的次数总和不过O(n),因此总的时间复杂度其实就是O(n)。

代码

class Solution {public:    vector<int> findDuplicates(vector<int>& nums) {        vector<int> ret;        for (int i = 0; i < nums.size(); ++i) {            while (nums[i] != i + 1) {                int value = nums[i];                if (value == 0) {                    break;                }                if (nums[value - 1] == nums[i]) {   // duplicates found                    ret.push_back(value);                    nums[value - 1] = 0;                }                nums[i] = nums[value - 1];                nums[value - 1] = value;            }        }        return ret;    }};