【剑指Offer】66.构建乘积数组（java）

最新推荐文章于 2022-12-14 11:58:53 发布

Q_smile

最新推荐文章于 2022-12-14 11:58:53 发布

阅读量218

点赞数 1

分类专栏： # 剑指offer

本文链接：https://blog.csdn.net/Q_smell/article/details/106027490

版权

剑指offer 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

LeetCode/剑指Offer/构建乘积数组

剑指Offer其他题目代码

给定一个数组 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]。

不能使用除法。

【思路】

这道题目的难点就在于不能使用除法，否则一个for就可以搞定了。

你有张良计，我有过墙梯。

一个for不行，那就多来几个，当然，不是嵌套在一起，不然就要超时了。

先来看一下B[i]的定义：

$B[i] = \prod_{u=0}^{i-1}A[u] *\prod_{u=i+1}^{n}A[u]$

很容易看出，i将A数组劈成了两半，那么，可以考虑用i这个位置去存储i位置之前元素的乘积，或者之后元素的乘积。

定义两个和A等长的数组，来搞定这件事。图解（因为比较大，就放在了资源里）

int[] b = new int[len];//存储i位置之前所有元素的乘积
b[0] = 1;
for (int i = 1; i < len; i++) {
    b[i] = b[i - 1] * a[i - 1];
}
        
int[] c = new int[len];//存储i位置之后所有元素的乘积
c[len - 1] = 1;        
for (int i = len - 2; i >= 0; i--) {
    c[i] = c[i + 1] * a[i + 1];
}

然后，将两个数组对应位置的元素相乘，便可得到答案。

用到了3个临时数组，跑了3个for循环，时空复杂度均为O(n).

【代码】

class Solution {
    public int[] constructArr(int[] a) {
        if (a.length == 0)
            return new int[0];
        int len = a.length;
        int[] b = new int[len];
        int[] c = new int[len];
        int[] res = new int[len];
        b[0] = 1;
        for (int i = 1; i < len; i++) {
            b[i] = b[i - 1] * a[i - 1];
        }
        c[len - 1] = 1;
        for (int i = len - 2; i >= 0; i--) {
            c[i] = c[i + 1] * a[i + 1];
        }

        for (int i = 0; i < len; i++) {
            res[i] = b[i] * c[i];
        }
        return res;
    }
}