开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第8天,点击查看活动详情
前言
本题为 LeetCode 前 100 高频题
我们社区陆续会将顾毅(Netflix 增长黑客,《iOS 面试之道》作者,ACE 职业健身教练。)的 Swift 算法题题解整理为文字版以方便大家学习与阅读。
LeetCode 算法到目前我们已经更新到 235 期,我们会保持更新时间和进度(周一、周三、周五早上 9:00 发布),每期的内容不多,我们希望大家可以在上班路上阅读,长久积累会有很大提升。
不积跬步,无以至千里;不积小流,无以成江海,Swift社区 伴你前行。如果大家有建议和意见欢迎在文末留言,我们会尽力满足大家的需求。
难度水平:中等
1. 描述
给你一个整数数组 nums,返回 数组 answer ,其中 answer[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积 。
题目数据 保证 数组 nums 之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内。
请不要使用除法,且在 O(n) 时间复杂度内完成此题。
2. 示例
示例 1
输入: nums = [1,2,3,4]
输出: [24,12,8,6]
示例 2
输入: nums = [-1,1,0,-3,3]
输出: [0,0,9,0,0]
提示:
2 <= nums.length <= 10^5-30 <= nums[i] <= 30- 保证数组
nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位整数范围内
进阶: 你可以在
O(1)的额外空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
3. 答案
方案一
class ProductExceptSelf {
func productExceptSelf(_ nums: [Int]) -> [Int] {
var zeroCount = 0
let total = nums.reduce(1) {
if $1 == 0 {
zeroCount += 1
}
return $0 * ($1 == 0 ? 1 : $1)
}
if zeroCount > 1 {return nums.map({_ in return 0})}
return nums.map({
if zeroCount == 1 {
return ($0 == 0 ? total : 0)
} else {
return ($0 == 0 ? total : total / $0)
}
})
}
}
- 主要思想:在迭代原始数组时,使用两个数组保存从左边和右边的乘法结果。
方案二
class ProductExceptSelfNotUseDivide{
func productExceptSelf(_ nums: [Int]) -> [Int] {
var arr = Array.init(repeating: 1, count: nums.count)
for i in (0..<(nums.count - 1)).reversed() {
arr[i] = arr[i + 1] * nums[i + 1]
}
var left = 1
for i in 0..<nums.count {
if i == 0 {
arr[i] = left * arr[i]
} else {
left = left * nums[i - 1]
arr[i] = left * arr[i]
}
}
return arr
}
}
-
主要思想:动态编程,同时跟踪左右产品列表,只使用一个额外的数组。问题陈述提到,使用答案数组不会增加空间复杂度
-
时间复杂度:O(n)
-
空间复杂度:O(1)
该算法题解的仓库:LeetCode-Swift
点击前往 LeetCode 练习
关于我们
我们是由 Swift 爱好者共同维护,我们会分享以 Swift 实战、SwiftUI、Swift 基础为核心的技术内容,也整理收集优秀的学习资料。