leetcode-剑指 Offer 66. 构建乘积数组
给定一个数组 A[0,1,…,n-1],请构建一个数组 B[0,1,…,n-1],其中 B[i] 的值是数组 A 中除了下标 i 以外的元素的积, 即 B[i]=A[0]×A[1]×…×A[i-1]×A[i+1]×…×A[n-1]。不能使用除法。
示例:
输入: [1,2,3,4,5]
输出: [120,60,40,30,24]
提示:
所有元素乘积之和不会溢出 32 位整数
a.length <= 100000
解题思路:
如果可以使用除法,直接将a所有的所有数字相乘,然后再遍历数组,对于每个数字结果就是总乘积除以每个数。这样做是很简单的,但是除法被限制啦。
那就换一个思路,首先遍历一遍数组,计算对于a[i]来讲,所有a[i]左边数字的乘积.
然后再遍历一遍数组,计算所有a[i]右边的乘积
最终,将a[i]左边的乘积 * 右边的乘积即可
class Solution {
public:
vector<int> constructArr(vector<int>& a) {
if(a.size()==0) return {};
vector<int> v1(a.size(),1);
vector<int> v2(a.size(),1);
for(int i=1;i<a.size();i++)
v1[i]=a[i-1]*v1[i-1];
for(int i=a.size()-2;i>=0;i--)
v2[i]=v2[i+1]*a[i+1];
for(int i=0;i<a.size();i++)
v2[i]*=v1[i];
return v2;
}
};
时间复杂度:O(n)
空间复杂度:O(n)