leetcode 238——除自身以外数组的乘积

题目：给定长度为 n 的整数数组 nums，其中 n > 1，返回输出数组 output ，其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。

示例:
  输入: [1,2,3,4]
  输出: [24,12,8,6]
说明: 请不要使用除法，且在 O(n) 时间复杂度内完成此题。

解题思路：两遍dp法

• dp1[i]保存[0,i)内元素的乘积，dp2[i]保存(i,size-1]内元素的乘积
• 然后output[i] = dp1[i] * dp2[i]

代码实现：

class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) {
        int size = nums.size();
        
        vector<int> dp1(size);
        vector<int> dp2(size);
        dp1[0] = 1;
        dp2[size-1] = 1;
        
        for (int i = 1; i < size; ++i) {
            dp1[i] = nums[i-1] * dp1[i-1];
        }
        for (int i = size-2; i >= 0; --i) {
            dp2[i] = nums[i+1] * dp2[i+1];
        }
        
        vector<int> res(size);
        for (int i = 0; i < size; ++i)
            res[i] = dp1[i] * dp2[i];
        
        return res;
    }
};