LintCode【简单】50. 数组剔除元素后的乘积 。代码及思路

题目要求：

给定一个整数数组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;
    }
};