LeetCode 0238 -- 除自身以外數組的乘積

除自身以外數組的乘積

題目描述

給定長度爲 n 的整數數組 nums，其中 n > 1，返回輸出數組 output ，其中 output[i] 等於 nums 中除 nums[i] 之外其餘各元素的乘積。

示例：

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

說明： 請不要使用除法，且在 O(n) 時間複雜度內完成此題。

進階：

你可以在常數空間複雜度內完成這個題目嗎？（ 出於對空間複雜度分析的目的，輸出數組不被視爲額外空間。）

解題思路

個人AC

從左往左遍歷： a0 = 1, a1 = a0, a2 = a0a1, a3 = a0a1a2

從右往左遍歷： a3 = 1, a2 = a3, a1 = a3a2, a0 = a3a2a1

累乘： a0 = a1a2a3, a1 = a0a2a3, a2 = a0a1a3, a3 = a0a1a2

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

時間複雜度： $O(n)$；

空間複雜度： $O(1)$。

最優解

同上。