入门客AI创业平台(我带你入门,你带我飞行)
博文笔记

Product of Array Except Self

创建时间:2015-09-16 投稿人: 浏览次数:345

Given an array of n integers where n > 1, nums, return an array output such that output[i] is equal to the product of all the elements of nums except nums[i].

Solve it without division and in O(n).

For example, given [1,2,3,4], return [24,12,8,6].

Follow up:
Could you solve it with constant space complexity? (Note: The output array does not count as extra space for the purpose of space complexity analysis.)

题目大意:

给定长度为n的整数数组nums,其中n > 1,返回输出数组output,满足output[i]等于除nums[i]之外其余各数的乘积。

不使用除法,在O(n)时间复杂度内完成此题目。

例如,给定 [1,2,3,4],返回 [24,12,8,6]。

进一步思考:

你可以在常数空间复杂度内完成题目吗?(注意:输出数组不算在空间复杂度分析中)

解题思路:

首先想到的思路是计算全部数字的乘积,然后分别除以num数组中的每一个数(需要排除数字0)。然而,题目要求不能使用除法。

下面的解法非常巧妙,参考LeetCode Dicuss

链接地址:https://leetcode.com/discuss/46104/simple-java-solution-in-o-n-without-extra-space

由于output[i] = (x 0 * x 1 * ... * x i-1 ) * (x i+1 * .... * x n-1 )

因此执行两趟循环:

第一趟正向遍历数组,计算x 0 ~ x i-1 的乘积

第二趟反向遍历数组,计算x i+1 ~ x n-1 的乘积

class Solution:
  # @param {integer[]} nums
  # @return {integer[]}
  def productExceptSelf(self, nums):
    size = len(nums)
    output = [1] * size
    left = 1
    for x in range(size - 1):
      left *= nums[x]
      output[x + 1] *= left
    right = 1
    for x in range(size - 1, 0, -1):
      right *= nums[x]
      output[x - 1] *= right
    return output


阅读更多
声明:该文观点仅代表作者本人,入门客AI创业平台信息发布平台仅提供信息存储空间服务,如有疑问请联系rumenke@qq.com。
  • 上一篇:没有了
  • 下一篇:没有了
未上传头像