(前缀和) LeetCode 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 位 整数范围内

二. 解题思路

本题意思是计算每一个numsi的值,其中numsi 的值为除自身以外的其他值的乘积。

首先我们定义两个数组 left 和 right ,其中 left 数组用来计算数组的前缀和,right 数组用来计算数组的后缀和,至于最终计算结果我们只需要在原数组上操作即可,省略了空间的浪费,numsi 的最终结果就等于numsi - 1 的前缀和乘 numsi + 1 的后缀和,即 numsi = lefti - 1 * righti + 1

但是在计算nums0 和numsn - 1 的时候我们发现会出现数组越界错误,所以我们将 left 数组元素统一后移一位,然后将 left0 赋予 1,将 right 数组扩展一位,rightn 赋予1 。所以就可以得出:numsi = lefti * righti + 1 (原本是 lefti - 1 * righti + 1,但是 left 元素统一后移一位,所以下标也会移动,但是 right 数组只是扩展,对下标未改动);

运算过程如图所示:

三. 代码

cpp 复制代码
class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) {
        int n = nums.size();
        int sum = 0;
        vector<int> left(n + 1);
        vector<int> right(n + 1);
        left[0] = 1;
        right[n] = 1;
        for(int i = 0; i < n; i++){
            left[i + 1] = nums[i] * left[i];
        }
        for(int i = n - 1; i >= 0; i--){
            right[i] = right[i + 1] * nums[i];
        }
        for(int i = 0; i < n; i++){
            nums[i] = left[i] * right[i + 1];
        }
        return nums;
    }
};

四. 总结

本题属于前缀和和后缀和的集合考察,属于中等题目,大家可以练习一下,但是一定要考虑在左右位置计算的时候的越界问题。

时间复杂度:O(n);

空间复杂度:O(n);

爱思考的小伙伴可以想一下本题如何用O(1)的空间复杂度实现,欢迎评论!

喜欢的话给个关注吧!!

相关推荐
Lsk_Smion9 分钟前
力扣实训 _ [200].岛屿数量
算法·leetcode·深度优先
Boom_Shu16 分钟前
长方形的关系
数据结构·c++·算法
ZhengEnCi1 小时前
O07-银行家算法
算法
装不满的克莱因瓶1 小时前
图像尺寸调整:缩放矩阵如何改变像素坐标?
人工智能·线性代数·数学·算法·机器学习·矩阵
思麟呀1 小时前
C++11并发编程:call_once一次性执行+atomic原子类型+CAS无锁编程+自旋锁
linux·开发语言·jvm·c++·windows
Lumbrologist1 小时前
【C++】零基础入门 · 第 13 节:类与对象基础
java·c++·算法
LONGZETECH1 小时前
软硬协同+故障注入:无人机仿真维修与操控仿真底层算法逻辑拆解
大数据·c语言·算法·3d·unity·无人机
Lsk_Smion1 小时前
力扣实训 _ [543].二叉树的直径 _ [23].合并K个升序列表
数据结构·算法·leetcode
凯瑟琳.奥古斯特2 小时前
力扣1235:加权区间调度最优解
java·python·算法·leetcode·职场和发展