题目描述 :
给定一个数组A[0,1,…,n-1],请构建一个数组B[0,1,…,n-1],其中B中的元素B[i]=A[0]* A[1]… A[i-1]* A[i+1]… A[n-1]。不能使用除法
这道题如果没有最后一个要求,直观想法就是将数组A中每个数相乘得出结果sum,每次求B[i]只需要sum / A[i]。但是题目不能使用除法,所以这种方法行不通,还有一点就是,如果存在某个A[i]为0,那么这个方法也需要改进,毕竟除数不能为0。
那么,对于这道题,可能现在想法就是依照定义求解,每次求B[i]都是取数组A中n-1个数相乘,最终就是n个B[i]时间复杂度为O(n* n)。一般来说 O(n*n)都不是最好的。
这道题有不少解法,关键点在于如何寻求数组中每个B[i]的关系,这里有个比较巧妙的方法
求B[i]就是求每行的乘积。
我们把它分成下三角,上三角,对于下三角,B[i] = B[i-1] * A[i-1],这样每次相乘都能使用上一次的值,因此最终复杂度为O(n),同理对于下三角也是这样处理。
class Solution {
public:
vector<int> multiply(const vector<int>& A)
{
int size = A.size();
vector<int> b(size, 1);
//求下三角
for (int i = 1; i < size; ++i)
{
b[i] = b[i-1] * A[i-1];
}
//求上三角
int temp = 1;
for (int i = size-2; i >= 0; --i)
{
temp *= A[i+1];
b[i] *= temp;
}
return b;
}
};