剑指 Offer 66. 构建乘积数组

最新推荐文章于 2023-04-09 20:57:16 发布

原创最新推荐文章于 2023-04-09 20:57:16 发布 · 104 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 专栏收录该内容

56 篇文章

订阅专栏

给定一个整数数组，构建一个新数组，其中每个元素是原数组中除当前下标外所有元素的乘积。该问题通过动态规划解决，分别从前往后和从后往前计算乘积，最后将两部分结果结合。示例给出了输入数组[1,2,3,4,5]的处理过程。

难度中等142

给定一个数组 A[0,1,…,n-1]，请构建一个数组 B[0,1,…,n-1]，其中 B[i] 的值是数组 A 中除了下标 i 以外的元素的积, 即 B[i]=A[0]×A[1]×…×A[i-1]×A[i+1]×…×A[n-1]。不能使用除法。

示例:

输入: [1,2,3,4,5]
输出: [120,60,40,30,24]

class Solution:
    def constructArr(self, a: List[int]) -> List[int]:
        n=len(a)
        dp=[1]*n
        tmp=1
        for i in range(1,n):
            dp[i]=dp[i-1]*a[i-1]  # 下三角
        for i in range(n-2,-1,-1):
            tmp=tmp*a[i+1]        # 上三角
            dp[i]=dp[i]*tmp  # 下三角 * 上三角
        return dp