剑指offer之构建乘积数组
问题描述
问题分析
- 题目意思很好理解,如果能够使用除法就很简单了,先算出A中各个元素的乘积s,然后B中对应元素就等于s除以对应的元素就行了
- 现在不能使用除法,一开始我的思路是:当遇到当前元素的时候让这个元素置为1,然后循环让A中每个元素相乘。这样很明显对每一个元素都要求一次A中每个元素的乘积,O(n2)的复杂度。这时候我就在想能否使用记忆化记录已经算出的乘积然后推导出B中的元素,哈哈,没想到怎么做。
- DP做法:首先要清楚,B中当前元素就是A中当前元素的左边所有元素的乘积和右边所有元素的乘积的乘积。想通这点,其实就是一个数组存着A中当前元素左边所有的元素的乘积,一个数组存A中当前元素右边所有的元素的乘积。然后二者相乘即可。
代码表述
暴力解法,超时。竟然过了43个用例,就一个例子没过。
class Solution {
public:
vector<int> constructArr(vector<int>& a) {
vector<int>res;
for(int i=0;i<a.size();i++){
int tmp=a[i];
a[i]=1;
int sum=1;
for(int j=0;j<a.size();j++){
sum*=a[j];
}
res.push_back(sum);
a[i]=tmp;
}
return res;
}
};
DP做法。
前缀积和后缀积??
class Solution {
public:
vector<int> constructArr(vector<int>& a) {
//当前元素的左边所有元素的乘积*当前元素的右边所有元素的乘积就是B数组位置对应的元素
//dp解法
int n=a.size();
vector<int>res(n,1);
vector<int>left(n,1);//保存当前元素的左边所有元素的乘积
vector<int>right(n,1);//保存当前元素的右边所有元素的乘积
for(int i=1;i<n;i++){
left[i]=left[i-1]*a[i-1];//乘积按照顺序的,所以由上一个元素得到的乘积乘上当前元素就好
}
for(int i=n-2;i>=0;i--){
right[i]=right[i+1]*a[i+1];//同理
}
for(int i=0;i<n;i++){
res[i]=left[i]*right[i];
}
return res;
}
};
时间复杂度:O(N)。三次for循环都要将数组遍历一遍
空间复杂度:O(N).额外的两个左右数组存储乘积。
可以直接在原数组上进行操作。
class Solution {
public:
vector<int> constructArr(vector<int>& a) {
//当前元素的左边所有元素的乘积*当前元素的右边所有元素的乘积就是B数组位置对应的元素
//dp解法
int n=a.size();
vector<int>res(n,1);
for(int i=1;i<n;i++){
res[i]=res[i-1]*a[i-1];
}
for(int i=n-2;i>=0;i--){
a[i]*=a[i+1];//直接更改原数组的元素值
res[i]=a[i+1]*res[i];//再次更改所求数组的元素值
}
return res;
}
};
时间复杂度:O(N)。for循环都要将数组遍历一遍
空间复杂度:O(1).没使用额外的数组。
复杂度分析
见上。