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

题目描述

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

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

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

思路

前缀积x后缀积。

代码

class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) {
        int n = nums.size();
        vector<int> answer(n);
        int* left=new int[n+1];
        int* right=new int[n+1];
        left[0]=1;
        right[n]=1;
        for(int i=1;i<=n;i++){
            left[i]=left[i-1]*nums[i-1];
            right[n-i]=right[n+1-i]*nums[n-i];
        }
        for(int i=0;i<n;i++){
            answer[i]=left[i]*right[i+1];
        }
        return answer;
    }
};
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

//空间O(1)优化解
class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) {
        int n = nums.size();
        vector<int> answer(n);
        int left = 1, right = 1;
        for (int i = 0; i < n; i++) {
            answer[i] = left;
            left *= nums[i];
        }
        for (int j = n - 1; j >= 0; j--) {
            answer[j] *= right;
            right *= nums[j];
        }
        return answer;
    }
};
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18