LeetCode-442 数组中重复的数据

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-all-duplicates-in-an-array

题目描述

给你一个长度为 n 的整数数组 nums ，其中 nums 的所有整数都在范围 [1, n] 内，且每个整数出现 一次 或 两次 。请你找出所有出现 两次 的整数，并以数组形式返回。

你必须设计并实现一个时间复杂度为 O(n) 且仅使用常量额外空间的算法解决此问题。

示例 1：

输入：nums = [4,3,2,7,8,2,3,1]
输出：[2,3]
示例 2：

输入：nums = [1,1,2]
输出：[1]
示例 3：

输入：nums = [1]
输出：[]

提示：

n == nums.length
1 <= n <= 10⁵
1 <= nums[i] <= n
nums 中的每个元素出现 一次 或 两次

解题思路

如果没有空间复杂度的需求，那么计数统计是很好的办法，但是题目要求空间复杂度为O(1)，那么不使用额外的空间去完成这道题，就需要在原数组中做文章了。由于数组的数据范围是1-n，如果不重复的话，那么每一个数字k将可以放入对应的k-1的位置，如果k-1位置上的不是k，那么可以判断这个数字重复。所以第一步，将数字k尽可能的放入对应的k-1位置上，第二步遍历数组寻找k-1位置上不是k的数字，加入结果。

代码展示

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

运行结果