leetcode 448. 找到所有数组中消失的数字(python)

题目链接

题目描述：

给定一个范围在 1 ≤ a[i] ≤ n ( n = 数组大小 ) 的 整型数组，数组中的元素一些出现了两次，另一些只出现一次。

找到所有在 [1, n] 范围之间没有出现在数组中的数字。

您能在不使用额外空间且时间复杂度为O(n)的情况下完成这个任务吗? 你可以假定返回的数组不算在额外空间内。

示例:

输入:
[4,3,2,7,8,2,3,1]

输出:
[5,6]

解题思路：
对数组进行排序，数组应该是递增的，每一个数比前一个数大1或者相等（重复数字），如果不满足这个条件，说明这两个数之间有不存在数组中的数字，将这部分数添加到返回数组中
由于缺失的数字可能是列表末位的数，没有一个标志将其添加到返回数组中，所以在遍历一次列表中进行判断，给定的数组排序后的最后一个数字是否等于数组长度，若不是，说明最后一个数字到数组长度之间的数字缺失，将其添加到返回数组中。

class Solution(object):
    def findDisappearedNumbers(self, nums):
        """
        :type nums: List[int]
        :rtype: List[int]
        """
        if len(nums)==0:
            return
        nums.sort()
        #print(nums)
        res=[]
        flag=0
        for i in range(1,len(nums)+1):
            if nums[i-1]==flag+1:
                flag=nums[i-1]
            elif nums[i-1]!=flag:
                for j in range(flag+1,nums[i-1]):
                    res.append(j)
                flag=nums[i-1]
        if nums[-1]!=len(nums):
            for k in range(nums[-1]+1,len(nums)+1):
                res.append(k)
        return res

别人的做法：
用一个数组h将nums中相应值的位置记为1，则h中为0的位置即为缺失的数字

        l = len(nums)+1
        h = [0]*l
        for i in nums:
            h[i]=1
        ans = []
        
        for i in range(1,l):
            if h[i]==0:
                ans.append(i)
        return ans