leetcode原题链接: 出自身以外数组的乘积
上一篇:HOT15-旋转数组
下一篇:HOT17-缺失的第一个正数
题目描述
给你一个整数数组 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]
解题方法: 累计和或者累计积,第一想到的是动态规划。假设left_dp[i]表示nums[i]左侧数字的乘积(不包括nums[i]), right_dp[i]表示nums[i]右侧数字的乘积(不包括nums[i]),则对于结果数组,result[k]=left[k]*right[k]。再针对这个可以做进一步的优化,可以用一个变量取代数组right_dp[i]的值。
C++代码
#include <iostream>
#include <vector>
class Solution {
public:
std::vector<int> productExceptSelf(std::vector<int>& nums) {
int n = nums.size();
std::vector<int> left(n, 1); //left[i]=nums[0]*nums[1]...*nums[i-2],表示nums[i]前面元素的乘积
for (int i = 1; i < n; i++) { //注意这里是<=,注意left[i]的含义
left[i] = left[i - 1] * nums[i - 1]; //这里容易写错,容易写成left[i]*=nums[i-1]
}
std::vector<int> result(n, 1); // result[i]表示第i+1个元素的结果值
int right = 1;//保存从右边向左边的乘积
for (int i = n - 1; i >= 0; i--) {
result[i] = left[i] * right; //先计算跟nums[i]右边的乘积
right *= nums[i];//后更新右边乘积的值
}
return result;
}
};