【LeetCode系列】Leet Code OJ 442. Find All Duplicates in an Array [Difficulty:Middle]

题目链接

这个题目，如果可以用空间复杂度O（n）的话，就是直接哈希表或者桶排序，如果要求时间复杂度O（nlogn）的话，那就直接排序好了，比如快速排序，归并排序或者堆排序。

但是，如果要求时间复杂度为O（n）又不创建任何空间的话，又该怎么办？

这时候，仔细看题就会发现：

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

题目的数据，数据范围是[1,n]，那么原本数组长度就是n,要实现低时间复杂度的话，不如用修改原本数组的方式来实现。

比如说对于数据:

[【4】,3,2,7,8,2,3,1]

现在，i在起始点，Key = nums[i]，就是4，那么我们把它和4这个位置替换一下

就变成了 [【7】,3,2,4,8,2,3,1]

然后这时候的值是7，同样的我们进行替换

[【3】,3,2,4,8,2,7,1]

[【2】,3,3,4,8,2,7,1]

[【3】,2,3,4,8,2,7,1]

这时候，key的值是3，但是nums[key]的值也是3，那么就代表3重复了，加进res数组中，然后把key置零，i往后移动

[0,2,3,4,8,2,7,1]

如果nums[i] = i + 1的话就继续往后移动,直到i为5时，key是8

[0，2，3，4，【8】，2，7，1]

[0，2，3，4，【1】，2，7，8]

[1，2，3，4，【0】，2，7，8]

这时候没有遇到什么重复 的情况，res不变，i继续往后移动，

[1，2，3，4，0，【2】，7，8]

因为2的位置已经是2了，因此，把2加入res数组，置零，i往后移动

[1，2，3，4，0，0，7，8]

这样就得到res的结果是[3,2](不用考虑顺序)

class Solution:
    def findDuplicates(self, nums):
        """
        :type nums: List[int]
        :rtype: List[int]
        """
        le = len(nums)
        res = []
        for i in range(le):
            if nums[i] - 1 == i: continue
            key = nums[i]
            while nums[key - 1] != key:
                tmp = nums[key - 1]
                nums[key - 1] = key
                nums[i] = tmp
                key = nums[i]
                if tmp == 0: break

            if nums[key - 1 ] == key and key - 1 != i and nums[i] != 0:
                res.append(key)
                nums[i] = 0
        return res