lintcode 数组剔除元素后的乘积
2016-12-29 本文已影响28人
yzawyx0220
给定一个整数数组A。
定义B[i] = A[0] * ... * A[i-1] * A[i+1] * ... * A[n-1], 计算B的时候请不要使用除法。
样例
给出A=[1, 2, 3],返回 B为[6, 3, 2]。
B[i]等于A[i]左边的数的乘积乘以A[i]右边的数的乘积,只需要知道这个这道题就非常简单了。
class Solution {
public:
/**
* @param A: 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> res(nums.size(),1);
long long temp = 1;
for (int i = 0;i < nums.size();i++) {
res[i] = temp;
temp *= nums[i];
}
temp = 1;
for (int i = nums.size()-1;i >= 0;i--) {
res[i] *= temp;
temp *= nums[i];
}
return res;
}
};