LeetCode - #238 除自身以外数组的乘积(Top 100)

83 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 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 基础为核心的技术内容,也整理收集优秀的学习资料。