LeetCode精选TOP面试题238.除自身以外数组的乘积

题目描述

• 给定一个整数数组 nums，返回 数组 answer ，其中 answer[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积 。

• 题目数据 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内。

• 不能使用除法，且在 O(n) 时间复杂度内完成此题。

example
input  : nums = [1,2,3,4]
output : [24,12,8,6]
input  : nums = [-1,1,0,-3,3]
output : [0,0,9,0,0]

解题思路

思路 左右遍历

• 从左向右遍历一次，对每个位置的元素，记录其左侧所有元素的乘积
• 再从右向左遍历一次，对每个位置的元素，在上一步乘积的基础上，累乘其右侧所有元素的乘积
• 时间复杂度O(n)
• 空间复杂度O(n)

代码（Java）

public class Solution {
    public int[] productExceptSelf(int[] nums) {
        int[] result = new int[nums.length];
        int sum = 1;
        result[0] = 1;
        for (int i = 0; i < nums.length - 1; i++) {
            sum *= nums[i];
            result[i + 1] = sum;
        }

        sum = 1;
        for (int i = nums.length - 1; i > 0; i--) {
            sum *= nums[i];
            result[i - 1] *= sum;
        }

        return result;
    }
}