力扣[238]除自身以外数组的乘积

题目

给你一个长度为 n 的整数数组 nums，其中 n > 1，返回输出数组 output ，其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。

示例:

输入: [1,2,3,4]
输出: [24,12,8,6]

提示：题目数据保证数组之中任意元素的全部前缀元素和后缀（甚至是整个数组）的乘积都在 32 位整数范围内。

说明: 请不要使用除法，且在 O(n) 时间复杂度内完成此题。

进阶：
你可以在常数空间复杂度内完成这个题目吗？（ 出于对空间复杂度分析的目的，输出数组不被视为额外空间。）

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/product-of-array-except-self
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

分析及题解

此题我最开的想法就是定义一个作为所有元素乘积的变量mul，然后遍历一次数组，得到所有元素的乘积，而对于output数组中的每个元素就应给是：
$$output[i]=mul/num[i]$$
但是！！！，题目中要求不允许使用除法，而且此种解法也会有许多复杂的情况需要分析，比如如果除数为0怎么办？如果num数组中有0的话，那岂不是最后mul值要为0？…………

再看题目中的提示，我们可以发现，对于数组中的元素num[i]，其除num[i]的元素乘积应为等于num[i]元素前的所有元素的乘积再乘上num[i]元素之后的所有元素的乘积，即定义num[i]元素前的所有元素的乘积为**num[i]的前缀乘积**，定义num[i]元素后的所有元素的乘积为**num[i]的后缀乘积**，则有：
$$除num[i]的元素乘积=num[i]的前缀乘积\times num[i]的后缀乘积$$
所以我们可以用l数组来存储每个元素的前缀乘积，用r数组来存储每个元素的后缀乘积，那么output数组中的元素即为l、r数组中对应元素的乘积，代码如下：

class Solution {
    public int[] productExceptSelf(int[] nums) {
        int[] l = new int[nums.length]; //前缀积
        int[] r = new int[nums.length]; //后缀积
        l[0] = 1; //首元素的前缀积定义为1
        r[nums.length - 1] = 1; //尾元素的后缀积定义为1
        for (int i = 1; i < nums.length; i++) {
            l[i] = nums[i - 1] * l[i - 1];
        }
        for (int i = nums.length - 2; i > -1; i--) {
            r[i] = nums[i + 1] * r[i + 1];
        }
        int[] output = new int[nums.length];
        for (int i = 0; i < nums.length; i++) {
            output[i] = l[i] * r[i];
        }
        return output;
    }
}

在这个代码中，我们对数组遍历了3次，分别生成了前缀积数组、后缀积数组和output数组，所以其时间复杂度为O(n)，题目中说明output数组不占空间复杂度，除output数组外我们开辟了前缀积、后缀积两个数组，所以我们的空间复杂度为O(n)，在题目的进阶要求中需要空间复杂度为O(1)，那我们能否将程序优化呢？

其实我们发现生成前缀积和后缀积的数组的目的就是为了得到output数组，而前缀积数组和后缀积数组其实并不是必要的，也就是说其实我们可以用output数组存储前缀积，然后再用临时变量表示相应位置的后缀积，再将临时变量乘上output数组中对应元素并将结果作为output数组的新值，最终的output数组即为结果，优化后**空间复杂度为O(1)**的代码如下：

class Solution {
    public int[] productExceptSelf(int[] nums) {
        int[] output = new int[nums.length];
        output[0] = 1;
        for (int i = 1; i < nums.length; i++) {
            output[i] = output[i - 1] * nums[i - 1]; //先将output数组作为前缀积数组
        }
        int r = 1;
        for (int i = nums.length - 1; i > -1; i--) {
            output[i] = output[i] * r;
            r = r * nums[i];
        }
        return output;
    }
}