Java之前缀积

前言

通过Java实现前缀积，这样可以不用除法完成除法的功能。

一、例题

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

二、暴力解

当然在leetcode通不过，仅供参考。当然也不建议，可以预热一下。
两层for循环来乘上所有数，除了自己。

1、源码

public int[] constructArr2(int[] a) {
        //从左到右计算每一位之前所有数的乘积
        int len = a.length;
        int[] res = new int[len];
        //双重for来计算,当遇到自己的时候就不要乘了
        for (int i = 0; i < len; i++) {
            res[i] = 1;
            for (int j = 0; j < len; j++) {
                if (i != j) {
                    res[i] *= a[j];
                }
            }
        }
        return res;
    }

三、前缀积

1）从左到右计算每一位之前所有数的乘积。
2）从右到左计算每一位之后所有数的乘积。
3）然后把两部分乘起来，刚好就没有乘上自己。

1、源码

public int[] constructArr(int[] a) {
        //从左到右计算每一位之前所有数的乘积
        int len = a.length;
        int[] res = new int[len];
        //之前的乘积，最左位之前的乘积初始化为1
        int product = 1;
        for (int i = 0; i < len; i++) {
            res[i] = product;
            product *= a[i];
        }
        //从右到左计算每一位之后的乘积，然后把两部分乘起来，刚好就没有乘上自己
        //之后的乘积，最右位之后的乘积初始化位1
        product = 1;
        for (int i = len - 1; i >= 0; i--) {
            res[i] *= product;
            product *= a[i];
        }
        return res;
    }

提供一个Leetcode的图来展示思路。

总结

1）暴力解，预热或特殊情况使用。
2）前缀积。

参考文献

[1] Leetcode原题之剑指offer2 66题
[2] 参考题解