Swift - LeetCode - 丢失的数字

题目

给定一个包含 [0, n]n 个数的数组 nums,找出 [0, n] 这个范围内没有出现在数组中的那个数。

示例 1:

  • 输入: nums = [3,0,1]
  • 输出: 2
  • 解释: n = 3,因为有 3 个数字,所以所有的数字都在范围 [0,3] 内。2 是丢失的数字,因为它没有出现在 nums 中。

示例 2:

  • 输入: nums = [0,1]
  • 输出: 2
  • 解释: n = 2,因为有 2 个数字,所以所有的数字都在范围 [0,2] 内。2 是丢失的数字,因为它没有出现在 nums 中。

示例 3:

  • 输入: nums = [9,6,4,2,3,5,7,0,1]
  • 输出: 8
  • 解释: n = 9,因为有 9 个数字,所以所有的数字都在范围 [0,9] 内。8 是丢失的数字,因为它没有出现在 nums 中。

示例 4:

  • 输入: nums = [0]
  • 输出: 1
  • 解释: n = 1,因为有 1 个数字,所以所有的数字都在范围 [0,1] 内。1 是丢失的数字,因为它没有出现在 nums 中。

方法一:排序

思路及解法

将数组排序之后,即可根据数组中每个下标处的元素是否和下标相等,得到丢失的数字。

由于数组的长度是 n,因此下标范围是 [0, n-1]。假设缺失的数字是 k,分别考虑以下两种情况:

  • 0 \le k < n 时,对任意 0 \le i < k,都有 \textit{nums}[i]=i,由于 k 缺失,因此 \textit{nums}[k]=k+1k 是第一个满足下标和元素不相等的下标;

  • k = n 时,0n-1 都没有缺失,因此对任意 0 \le i < n,都有 \textit{nums}[i]=i

根据上述两种情况,可以得到如下方法得到丢失的数字:

  1. 从左到右遍历数组 \textit{nums},如果存在 0 \le i < n 使得 \textit{nums}[i] \ne i,则缺失的数字是满足 \textit{nums}[i] \ne i 的最小的 i

  2. 如果对任意 0 \le i < n,都有 \textit{nums}[i]=i,则缺失的数字是 n

代码

class Solution {
    func missingNumber(_ nums: [Int]) -> Int {
        let tempNums: [Int] = nums.sorted()
        for i in 0..<tempNums.count {
            if tempNums[i] != i {
                return i
            }
        }
        return tempNums.count
    }
}

复杂度分析

  • 时间复杂度:O(nlogn),其中 n 是数组 \textit{nums} 的长度。排序的时间复杂度是 O(n \log n),遍历数组寻找丢失的数字的时间复杂度是 O(n),因此总时间复杂度是 O(n \log n)

  • 空间复杂度:O(logn),其中 n 是数组 \textit{nums} 的长度。空间复杂度主要取决于排序的递归调用栈空间。

方法二:哈希集合

思路及解法

使用哈希集合,可以将时间复杂度降低到 O(n)

首先遍历数组 \textit{nums},将数组中的每个元素加入哈希集合,然后依次检查从 0n 的每个整数是否在哈希集合中,不在哈希集合中的数字即为丢失的数字。由于哈希集合的每次添加元素和查找元素的时间复杂度都是 O(1),因此总时间复杂度是 O(n)

代码

class Solution {
    func missingNumber(_ nums: [Int]) -> Int {
        let set: Set<Int> = Set(nums)
        for i in 0...nums.count {
            if !set.contains(i) {
                return i
            }
        }
        return -1
    }
}

复杂度分析

  • 时间复杂度:O(n),其中 n 是数组 \textit{nums} 的长度。遍历数组 \textit{nums} 将元素加入哈希集合的时间复杂度是 O(n),遍历从 0n 的每个整数并判断是否在哈希集合中的时间复杂度也是 O(n)

  • 空间复杂度:O(n),其中 n 是数组 \textit{nums} 的长度。哈希集合中需要存储 n 个整数。

方法三:位运算

思路及解法

数组 \textit{nums} 中有 n 个数,在这 n 个数的后面添加从 0n 的每个整数,则添加了 n+1 个整数,共有 2n+1 个整数。

2n+1 个整数中,丢失的数字只在后面 n+1 个整数中出现一次,其余的数字在前面 n 个整数中(即数组中)和后面 n+1 个整数中各出现一次,即其余的数字都出现了两次。

根据出现的次数的奇偶性,可以使用按位异或运算得到丢失的数字。按位异或运算 \oplus⊕ 满足交换律和结合律,且对任意整数 x 都满足 x \oplus x = 0x \oplus 0 = x

由于上述 2n+1 个整数中,丢失的数字出现了一次,其余的数字都出现了两次,因此对上述 2n+1 个整数进行按位异或运算,结果即为丢失的数字。

代码

class Solution {
    func missingNumber(_ nums: [Int]) -> Int {
        var res = 0
        for i in 0..<nums.count {
            res ^= nums[i]
        }
        for i in 0...nums.count {
            res ^= i
        }
        return res
    }
}

复杂度分析

  • 时间复杂度:O(n),其中 n 是数组 \textit{nums} 的长度。需要对 2n+1 个数字计算按位异或的结果。

  • 空间复杂度:O(1)

方法四:数学

思路及解法

将从 0n 的全部整数之和记为 \textit{total},根据高斯求和公式,有:

\textit{total} = \sum_{i=1}^n = \dfrac{n(n+1)}{2}

将数组 \textit{nums} 的元素之和记为 \textit{arrSum},则 \textit{arrSum}\textit{total} 少了丢失的一个数字,因此丢失的数字即为 \textit{total}\textit{arrSum} 之差。

代码

class Solution {
    func missingNumber(_ nums: [Int]) -> Int {
        var n: Int = nums.count
        var total: Int = n * (n + 1) / 2
        for i in 0..<nums.count {
            total -= nums[i]
        }
        return total
    }
}

复杂度分析

  • 时间复杂度:O(n),其中 n 是数组 \textit{nums} 的长度。需要 O(1) 的时间计算从 0n 的全部整数之和以及需要 O(n) 的时间计算数组 \textit{nums} 的元素之和。

  • 空间复杂度:O(1)

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容