【数据结构基础_数组】Leetcode 238.除自身以外数组的乘积

最新推荐文章于 2024-09-15 15:29:37 发布

cwtnice

最新推荐文章于 2024-09-15 15:29:37 发布

阅读量87

点赞数

分类专栏：每日一题文章标签： leetcode

本文链接：https://blog.csdn.net/cwtnice/article/details/126051351

版权

每日一题专栏收录该内容

118 篇文章 2 订阅

订阅专栏

本文介绍了如何在O(n)时间和O(1)额外空间复杂度下解决LeetCode第238题——产品数组除自身。方法一是通过维护左右乘积数组，分别计算每个元素左侧和右侧的乘积，然后两者相乘得到答案；方法二是进一步优化空间复杂度，不使用额外数组。

摘要由CSDN通过智能技术生成

原题链接：Leetcode 238. Product of Array Except Self

Given an integer array nums, return an array answer such that answer[i] is equal to the product of all the elements of nums except nums[i].

The product of any prefix or suffix of nums is guaranteed to fit in a 32-bit integer.

You must write an algorithm that runs in O(n) time and without using the division operation.

Example 1:

Input: nums = [1,2,3,4]
Output: [24,12,8,6]

Example 2:

Input: nums = [-1,1,0,-3,3]
Output: [0,0,9,0,0]

Constraints:

2 <= nums.length <= 10⁵
-30 <= nums[i] <= 30
The product of any prefix or suffix of nums is guaranteed to fit in a 32-bit integer.

Follow up:

Can you solve the problem in O(1) extra space complexity? (The output array does not count as extra space for space complexity analysis.)

方法一：左右乘积数组

思路：

维护两个数组l和r
其中l[i]表示nums[i]左边所有元素的乘积和，r[i]表示nums[i]右边所有元素的乘积和
那么l[0] = r[n-1] = 1，因为他们的外边并没有元素
最后通过左右两个乘积再求乘积得到答案

C++代码：

class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) {
        int n = nums.size();

        // 左右乘积列表 l(i)为i左侧所有元素的乘积 r(i)为i右侧所有元素的乘积
        vector<int> l(n, 0);
        vector<int> r(n, 0);
        vector<int> ans(n, 0);

        // 维护两个列表 最开头两个值为1
        l[0] = r[n-1] = 1;
        for(int i = 1; i < n; i++ )
            l[i] = l[i - 1] * nums[i - 1];
        for(int j = n - 2; j >= 0; j-- )
            r[j] = r[j + 1] * nums[j + 1];

        for(int i = 0; i < n; i++ )
            ans[i] = l[i] * r[i];
        return ans;
    }
};