题目要求:
给定一个整数数组A。
定义B[i] = A[0] * ... * A[i-1] * A[i+1] * ... * A[n-1], 计算B的时候请不要使用除法。
样例
给出A=[1, 2, 3],返回 B为[6, 3, 2]
思路:
根据给出的样例,很容易发现当i不是两端的时候,b[i]等于a[0]到a[i-1]的乘积乘上a[i+1]到a[n-1]的乘积。
当i在两端时,把两端的值变为1,再乘上另一端的值就好。
但是我想,当a只有一个[0]时,b[0]应该是0才对,但LintCode的编译出来是1……这个应该是个错误吧。
代码:
class Solution {
public:
/*
* @param nums: Given an integers array A
* @return: A long long array B and B[i]= A[0] * ... * A[i-1] * A[i+1] * ... * A[n-1]
*/
vector<long long> productExcludeItself(vector<int> &nums) {
// write your code here
vector<long long> b;
int i, j;
long long front, rear, sum;
for(i = 0; i < nums.size(); i++){
front = 1, rear = 1;
for(j = 0; j < nums.size(); j++){
if(j <= i - 1) front *= nums[j];
if(j >= i + 1) rear *= nums[j];
}
sum = front * rear;
b.push_back(sum);
}
return b;
}
};