思路:我们可以这样想举个维度是3的栗子:
1 | a1 | a2 |
a0 | 1 | a2 |
a1 | a0 | 1 |
那我们用两个数组把左右部分分割开,数组存储的分别是对应一半的一行中的乘积,于是我们可以很容易的用迭代循环的方式求出两个数组中的值
比如两个数组left[],right[],在上面的例子中就是left[3]={1,a0,a0*a1},right[3]={a1*a2,a2,1}
然后把对应部分乘起来返回就好了。
代码如下所示:
class Solution {
public:
vector<int> constructArr(vector<int>& a) {
int len=a.size();
vector<int> left(len,1),right(len,1),res;
for(int i=1;i<len;++i)
{
left[i]=left[i-1]*a[i-1];
right[len-i-1]=right[len-i]*a[len-i];
}
for(int i=0;i<len;++i)
{
res.emplace_back(left[i]*right[i]);
}
return res;
}
};
结果如下所示:
当然也可以优化以下存储啥的:
class Solution {
public:
vector<int> constructArr(vector<int>& a) {
int len=a.size();
vector<int> left(len,1),right(len,1);
// vector<int> res;
for(int i=1;i<len;++i)
{
left[i]=left[i-1]*a[i-1];
right[len-i-1]=right[len-i]*a[len-i];
}
for(int i=0;i<len;++i)
{
right[i]*=left[i];
// res.emplace_back(left[i]*right[i]);
}
// return res;
return right;
}
};
结果如下所示:
为什么时间提升了。。。不懂