【LeetCode】Product of Array Except Self

标签：

Product of Array Except Self

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.)

Solution:

题目中说不要用到 division 并且要在线性时间内得出结果。

首先受上一题影响想想能不能位运算，但这题乘法是不大行的。

那么要如果扫描一遍就得出要求的Array呢？

想想我们遍历 Array 时我们可以干些什么，我们可以对当前秩所在位置进行操作，关键是我们还可以对当前秩前后常数位置也可以进行操作。

考虑当前遍历到的秩 i，我们可以把得出 ans[i] 的过程分为两部分：一部分是对于所有小于 i 的秩的元素的相乘，另一部分则是所有大于 i 的秩的元素的相乘。

独立地来完成这两部分并不难，只要在遍历一遍时，用一个变量记录已经遍历到的元素的乘积（不包括当前元素），乘到 ans[i]即可。

而这两个独立的部分其实也可以合成一个遍历来完成，因为只要知道了整个需要遍历的Array的长度，从左向右和从右向左只是一个取模的问题了。

代码如下：

 1 class Solution:
 2     # @param {integer[]} nums
 3     # @return {integer[]}
 4     def productExceptSelf(self, nums):
 5         n = len(nums)
 6         ans = [1] * n
 7         left_fac, right_fac = 1, 1
 8         for i in range(n):
 9             ans[i] *= left_fac
10             left_fac *= nums[i]
11             ans[n - i - 1] *= right_fac
12             right_fac *= nums[n - i - 1]
13         return ans

【LeetCode】Product of Array Except Self

标签：

原文地址：http://www.cnblogs.com/maples7/p/4725793.html