力扣 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

思路：前缀和，后缀和。这两者的结合。

时间复杂度O(N)，空间复杂度O(N)。

代码

    public int[] productExceptSelf(int[] nums) {
        int length = nums.length;
        System.out.println(length);
        int []pre = new int[length + 5];
        int []post = new int[length + 5];
        int []ans = new int[length];
        // 前缀和
        for (int i = 0; i < length; i++) {
            if (i == 0) {
                pre[i] = nums[i];
            } else{
                pre[i] = pre[i - 1] * nums[i];
            }
        }
        // 后缀和
        for (int i = length -1; i >= 0; i--) {
            if (i == length - 1) {
                post[i] = nums[i];
            } else{
                post[i] = post[i + 1] * nums[i];
            }
        }
        // 结果数组
        for (int i = 0; i < length; i++) {
            if (i == 0) {
                ans[i] = post[i + 1];
            } else if(i == length - 1) {
                ans[i] = pre[i -1];
            } else{
                ans[i] = pre[i-1] * post[i + 1];
            }
        }
        return ans;
    }