题目描述
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/product-of-array-except-self
给你一个长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
示例:
输入: [1,2,3,4]
输出: [24,12,8,6]
提示:题目数据保证数组之中任意元素的全部前缀元素和后缀(甚至是整个数组)的乘积都在 32 位整数范围内。
说明: 请不要使用除法,且在 O(n) 时间复杂度内完成此题。
进阶:
你可以在常数空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
思路
题目要求在 O(n) 时间复杂度内完成此题 , 且不能用除法。
那么我们可以用使用前缀后缀的思想 用一个数组保存num中每个数的前缀积 , 再用一个数组保存每个数的后缀积
如例子中[1 2 3 4]
对应的前缀积就是[1,1,2,6]后缀积就是[24,12,4,1]
那么每个数对应的前缀积 * 后缀积就是除了这个数之外所有数的积
则[1 , 1, 2, 6]
- [24,12,4,1]
= [24 , 12, 8, 6] 也就是答案啦;
用一个for循环计算前缀积 一个for 循环计算前缀积 一个for循环计算前缀积*后缀积
时间复杂度为O(0) 就满足题目要求啦
代码
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int length = nums.size();
vector<int> L(length,0);
vector<int> R(length,0);
vector<int> res(length);
L[0] = 1;
for(int i = 1; i < length; i++)
{
L[i] = L[i-1] * nums[i-1];
}
R[length - 1] = 1;
for(int i = length - 2; i >= 0; i--)
{
R[i] = R[i + 1] * nums [i + 1];
}
for(int i = 0; i < length; i++)
{
res[i] = L[i] * R[i];
}
return res;
}
};