238. Product of Array Except Self

除了当前元素之外的元素相乘。

要求:不能用除法

难度:【medium】

for ex.

Input:  [1,2,3,4]
Output: [24,12,8,6]

// 24 = 2*3*4
// 12 = 1*3*4
// 8 = 1*2*4
// 6 = 1*2*3
class Solution {
public:
    vector productExceptSelf(vector& nums) {
        vector res(nums.size());
               
        res[0] = 1;
        // 先用res保存每个元素左边元素相乘的结果
        // res[1] 是nums[0]
        // res[2] 是nums[0]*nums[1]
        // res[3] 是nums[0]*nums[1]*nums[2]
        // res[n - 1] 是nums[0]*nums[1]*...*nums[n - 2]
        for (int i = 1; i < nums.size(); ++i) {
            res[i] = nums[i - 1] * res[i - 1];
        }
        
        long product = 1;
        // product保存每个元素右边元素乘积
        for (int i = nums.size() - 1; i >= 1; --i) {
            product *= nums[i];
            res[i - 1] *= product; // 左边元素乘积 * 右边元素乘积
        }
        
        return res;
    }
};

 

你可能感兴趣的:(leetcode,array)