LeetCode算法刷题——238. 除自身以外数组的乘积

一、题目描述

二、解题思路

我们可以使用前缀积和后缀积的方法来解决这个问题,避免使用除法:

  1. 对于每个位置 i,我们需要计算 numsi 左侧 所有元素的乘积和右侧所有元素的乘积

  2. 先从左到右遍历,计算每个位置左侧所有元素的乘积,存入 answer 数组

  3. 再从右到左遍历,计算每个位置右侧所有元素的乘积,并与左侧乘积相乘得到最终结果

  4. 这样只需要两次遍历,时间复杂度 O(n),空间复杂度 O(1)(不考虑输出数组)

三、完整代码

class Solution {

public:

vector<int> productExceptSelf(vector<int>& nums) {

// 初始化结果数组,大小为 nums.size(),所有元素初始化为1

vector<int> answer(nums.size(), 1);

int sum = 1; // 用于记录右侧乘积

// 第一遍遍历:计算每个位置左侧所有元素的乘积

for (int i = 1; i < nums.size(); i++) {

answeri = answeri - 1 * numsi - 1;

}

// 第二遍遍历:计算每个位置右侧所有元素的乘积,并与左侧乘积相乘

for (int i = nums.size() - 2; i >= 0; i--) {

sum = sum * numsi + 1;

answeri = sum * answeri;

}

return answer;

}

};

四、代码解析

1. 初始化结果数组

vector<int> answer(nums.size(), 1);

  • 创建大小为 nums.size() 的向量

  • 所有元素初始化为 1,因为乘法中 1 是单位元

2. 计算左侧乘积

for (int i = 1; i < nums.size(); i++) {

answeri = answeri - 1 * numsi - 1;

}

  • 从索引 1 开始遍历**(索引 0 的左侧乘积初始化时已经设置为 1)**

  • answeri 存储 numsi 左侧所有元素的乘积

  • 递推关系:第 i 个位置的左侧乘积 = 第 i-1 个位置的左侧乘积 × numsi-1

3. 计算右侧乘积并更新结果

for (int i = nums.size() - 2; i >= 0; i--) {

sum = sum * numsi + 1;

answeri = sum * answeri;

}

  • 从倒数第二个元素开始向前遍历**(最后一个索引 的右侧乘积初始化时已经设置为 1)**

  • sum 变量累乘右侧元素,初始为 1

  • 对于位置 i,sum 存储 numsi 右侧所有元素的乘积

  • 将左侧乘积和右侧乘积相乘得到最终结果

五、语法要点

1. vector初始化大小和数值

vector<int> v(5, 1); // 创建包含5个元素的向量,所有元素初始化为1

2. vector下标赋值的限制

vector<int> v; // 创建空向量

v0 = 1; // 错误 ❌

v.push_back(1) // 正确 ✅

  • 未初始化的向量不能直接通过下标赋值,只能使用push_back或push_front方法在前后插入!

六、执行示例

输入:nums = [1,2,3,4]

第一遍遍历(计算左侧乘积):

  • i=0: answer0 = 1(初始化值)

  • i=1: answer1 = answer0 × nums0 = 1 × 1 = 1

  • i=2: answer2 = answer1 × nums1 = 1 × 2 = 2

  • i=3: answer3 = answer2 × nums2 = 2 × 3 = 6

  • 此时 answer = 1, 1, 2, 6

第二遍遍历(计算右侧乘积并更新):

  • 初始化 sum = 1

  • i=2: sum = 1 × nums3 = 1 × 4 = 4, answer2 = 4 × 2 = 8

  • i=1: sum = 4 × nums2 = 4 × 3 = 12, answer1 = 12 × 1 = 12

  • i=0: sum = 12 × nums1 = 12 × 2 = 24, answer0 = 24 × 1 = 24

  • 最终 answer = 24, 12, 8, 6

七、总结

本文介绍了计算除自身以外数组乘积的高效算法。通过分别计算每个元素的左侧乘积和右侧乘积,然后将两者相乘,我们可以在 O(n) 时间复杂度内解决问题,且不需要使用除法。算法的关键在于两次遍历:第一次计算左侧乘积,第二次计算右侧乘积并更新结果。这种方法空间复杂度为 O(1)(不考虑输出数组),是解决此类问题的最优解法。同时,我们学习了 vector 的正确初始化方法,避免直接通过下标访问未初始化的向量元素。

相关推荐
xier_ran1 小时前
【infra之路】AI 编译器学习笔记(二):重拾 C++——移动语义与 LLVM 风格的类型转换
人工智能·笔记·学习
Nil2082 小时前
leetcode 198打家劫舍
算法·leetcode·职场和发展
Shaoshing3 小时前
前缀和与差分
java·数据结构·算法·前缀和·差分
mmmmath_33 小时前
LeetCode.429.N叉树的层序遍历
数据结构·leetcode
li星野3 小时前
5 TYPES OF ADVERBS — One-Page Notes
学习
星恒随风3 小时前
Linux库的底层原理:ELF文件格式、静态链接、程序加载、虚拟地址空间与GOT/PLT详解
linux·运维·笔记·学习·状态模式
罗西的思考3 小时前
[Agent Memory / 强化学习] MemPO源码学习笔记 — (7)— 混合机制
笔记·学习
loulanyue_3 小时前
构建房产认知决策智能:Harness底座与AI自进化的技术实践——读贝壳首席算法科学家尹俊希2026云栖专场演讲
人工智能·算法
碧海银沙音频科技研究院4 小时前
基于炬芯ATS录音卡BLE控制协议
人工智能·嵌入式硬件·算法·语音识别
jimy14 小时前
C++ 虚函数表:基类和派生类共用一张 vtable 吗?
开发语言·c++