构建乘积数组

题目描述

给定一个数组 A[0,1,…,n-1]，请构建一个数组 B[0,1,…,n-1]，其中 B 中的元素 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]

思路

1.可以从题目给出的公式中，推导出一幅图。

2. 根据图片的内容，可以通过计算“三角形”的结果，推导出最终元素的结果。

3.三角形主要包括，上半部和下半部，具体计算详情，可以参看代码。

Java代码实现

class Solution {
    public int[] constructArr(int[] a) {
        if(a.length == 0){
            return a;
        }
        int[] res = new int[a.length];
        res[0] = 1;

        //计算下三角
        for (int i = 1; i < res.length; i++) {
            res[i] = res[i-1] * a[i-1];
        }
        
        int temp = 1;

        //计算上三角
        for (int i = res.length-2; i >=0 ; i--) {
            temp *= a[i+1];
            res[i] *= temp;
        }
        
        return res;
    }
}

Golang代码实现

func constructArr(a []int) []int {
    if len(a) == 0{
        return a
    }
    res := make([]int,len(a))
    res[0] = 1
    //计算上三角
    for i:=1; i<len(a); i++{
        res[i] = res[i-1] * a[i-1]
    }
    temp := 1
    
    //计算下三角
    for i:=len(a)-2; i>=0; i--{
        temp *= a[i+1]
        res[i] *= temp
    }
    
    return res
}

有疑问加站长微信联系（非本文作者）