题目描述
给定一个数组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]。不能使用除法。(注意:规定B[0] = A[1] * A[2] * … * A[n-1],B[n-1] = A[0] * A[1] * … * A[n-2];)
思路:
按照二维坐标的方式计算上三角和下三角。
代码:
class Solution {
public:
vector<int> multiply(const vector<int>& A) {
int n=A.size();
vector<int> res(n); //结果数组。
if(n<1)
return res;
res[0]=1;
//开始计算上三角
for(int i=1; i<n; i++)//第一行上三角没有,所以从第二行1开始
{
res[i]=res[i-1]*A[i-1];
}
//开始右边的下三角
int tmp=1;
for(int i=n-2;i>=0;i--)
{
tmp=A[i+1]; //n-2行没有n-2,是从n-1开始的
res[i]=res[i]*tmp;
}
return res;
}