列表缺失的正整数

Find Missing Numbers in an Array

列表缺失的正整数

问题描述
给定一个有 n 的元素的正整数列表，列表中的元素是 1 到 n 的正整数，找到 [1, n] 区间中缺少的正整数，列表中可能出现相同的数值。

测试样例

# Input: 
nums = [4, 6, 2, 6, 7, 2, 1]
# Output:
[3, 5]
# 列表中有 7 个元素，列表中包含的正整数有 1，2，4，5，6，7。
# 缺少 3 和 5.

内容首发于微信公众号IT信息教室，如果您想学习更多AI相关的技能，欢迎搜索关注或微信扫描下方二维码关注～～

参考代码

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# ---------------------
# # O(n) time｜O(1) space
# ---------------------
class Solution:
    def findDisappearedNumbers(self, nums):
        for idx in range(len(nums)):
            index = abs(nums[idx]) - 1
            nums[index] = -abs(nums[index])
        return [idx for idx, num in enumerate(nums, start = 1) if num > 0]
    
# Test Program
nums = [4, 6, 2, 6, 7, 2, 1]
result = Solution().findDisappearedNumbers(nums)
print(result)
print(''.join(str(result).split()))
# [3, 5]
# [3,5]