前缀和(四)除自身以外数组的乘积

238. 除自身以外数组的乘积

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

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

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

示例 1:

复制代码
输入: nums = [1,2,3,4]
输出: [24,12,8,6]

示例 2:

复制代码
输入: nums = [-1,1,0,-3,3]
输出: [0,0,9,0,0]

提示:

  • 2 <= nums.length <= 105
  • -30 <= nums[i] <= 30
  • 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内

对于每⼀个位置的最终结果 ret[i] ,它是由两部分组成的:

i. nums[0] * nums[1] * nums[2] * ... * nums[i - 1]

ii. nums[i + 1] * nums[i + 2] * ... * nums[n - 1]

于是,我们可以利⽤前缀和的思想,使⽤两个数组 left 和 right,分别处理出来两个信息:

i. left 表⽰:i 位置之前的所有元素,即 [0, i - 1] 区间内所有元素的前缀乘积,

ii. right表⽰: i 位置之后的所有元素,即 [i + 1, n - 1] 区间内所有元素的后缀乘积

然后再处理最终结果

cpp 复制代码
res[i-1] = left[i-1] * right[i+1];
cpp 复制代码
class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) {
        int n = nums.size();
        vector<int> left(n+1);  // 注意left初始化个数
        vector<int> right(n+2); // 注意right初始化个数 
        left[0] = 1, right[n+1] = 1;
        for(int i = 1; i <= n; i++)
            left[i] = left[i-1] * nums[i-1];    // 下标访问
        for(int i = n; i >= 1; i--)
            right[i] = right[i+1] * nums[i-1];
        
        vector<int> res(n);
        for(int i = 1; i <= n; i++)
        {
            res[i-1] = left[i-1] * right[i+1];
        }
        return res;
    }
};
相关推荐
虾..5 分钟前
Linux 简单日志程序
linux·运维·算法
Trent198513 分钟前
影楼精修-眼镜祛反光算法详解
图像处理·人工智能·算法·计算机视觉·aigc
蓝色汪洋14 分钟前
经典修路问题
开发语言·c++·算法
csuzhucong25 分钟前
122魔方、123魔方
算法
Salt_072838 分钟前
DAY 40 早停策略和模型权重的保存
人工智能·python·算法·机器学习
卜锦元1 小时前
Golang后端性能优化手册(第三章:代码层面性能优化)
开发语言·数据结构·后端·算法·性能优化·golang
Binky6781 小时前
力扣--回溯篇(2)
算法·leetcode·职场和发展
De-Alf2 小时前
Megatron-LM学习笔记(6)Megatron Model Attention注意力与MLA
笔记·学习·算法·ai
2401_841495642 小时前
【LeetCode刷题】打家劫舍
数据结构·python·算法·leetcode·动态规划·数组·传统dp数组
冰西瓜6002 小时前
STL——vector
数据结构·c++·算法