【LintCode 简单】50. 数组剔除元素后的乘积

最新推荐文章于 2018-06-02 10:58:24 发布

麦片加奶不加糖

最新推荐文章于 2018-06-02 10:58:24 发布

阅读量172

点赞数

分类专栏： # LintCode 文章标签： LintCode Python 数组

本文链接：https://blog.csdn.net/sinat_23133783/article/details/79063723

版权

LintCode 专栏收录该内容

90 篇文章 1 订阅

订阅专栏

1.问题描述：

给定一个整数数组A。

定义B[i] = A[0] * ... * A[i-1] * A[i+1] * ... * A[n-1]，计算B的时候请不要使用除法。

2.样例:

给出A=[1, 2, 3]，返回 B为[6, 3, 2]

3.代码：

class Solution:
    """
    @param: nums: Given an integers array A
    @return: A long long array B and B[i]= A[0] * ... * A[i-1] * A[i+1] * ... * A[n-1]
    """
    def productExcludeItself(self, nums):
        # write your code here
        n = len(nums)
        B=[]
        accumulate = 1
        count=0
        for i in range(n):
            if nums[i]!=0:
                accumulate = accumulate * nums[i]
            else:
                count+=1
        # no zero in array A
        if count==0:
            for j in range(n):
	            B.append(int(accumulate / nums[j]))
        else:
            for j in range(n):
                # only one zero
                if nums[j]==0 and count==1:
                    B.append(accumulate)
                # more than one zero
                else:
                    B.append(0)
        return B

class Solution:
    def productExcludeItself(self, A):
        length ,B  = len(A) ,[]
        f = [ 0 for i in xrange(length + 1)]
        f[ length ] = 1
        for i in xrange(length - 1 , 0 , -1):
            f[ i ] = f[ i + 1 ] * A[ i ]
        tmp = 1
        for i in xrange(length):
            B.append(tmp * f[ i + 1 ])
            tmp *= A[ i ]
        return B