题目描述有问题,给出的A应该是{1,2,3,4,5}.而不是从0开始,给出的测试用例也是从一开始,这点很坑。
这道题就是考察边界条件的分析以及数组找规律。
B的每一行分为两部分计算,以对角线为分割,C为左上三角数组,D为右下三角数组,B = C * D;
vector<int> multiply(const vector<int>& A) {
int size = A.size();
vector<int> B(size);
vector<int> C(size);
vector<int> D(size);
C[0] = 1;
for (int i = 1; i < size; i++) {
C[i] = C[i - 1] * A[i - 1];
}
D[size - 1] = 1;
for (int i = size - 2; i >= 0; i--) {
D[i] = D[i + 1] * A[i+1];
}
for (int i = 0; i < size; i++) {
B[i] = C[i] * D[i];
}
return B;
}