【LeetCode】238.除自身意外数组的乘积

问题描述

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

示例:

输入: [1,2,3,4]
输出: [24,12,8,6]

提示：题目数据保证数组之中任意元素的全部前缀元素和后缀（甚至是整个数组）的乘积都在 32 位整数范围内。

说明: 请不要使用除法，且在 O(n) 时间复杂度内完成此题。

进阶：你可以在常数空间复杂度内完成这个题目吗？（ 出于对空间复杂度分析的目的，输出数组不被视为额外空间。）

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/product-of-array-except-self
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

法I：动态规划（左右乘积列表）——时间复杂度O(n)，空间复杂度O(n)

class Solution {
public:
	vector<int> productExceptSelf(vector<int>& nums) {
		vector<int> output(nums.size());
		vector<int> beforProduct(nums.size());//[0,i)
		vector<int> afterProduct(nums.size());//(i,nums.size()-1]
		beforProduct[0] = 1;
		for (int i = 1; i < nums.size(); i++) {
			beforProduct[i] = beforProduct[i - 1] * nums[i - 1];
		}
		afterProduct[nums.size() - 1] = 1;
		for (int i = nums.size() - 2; i >= 0; i--) {
			afterProduct[i] = afterProduct[i + 1] * nums[i + 1];
		}
		for (int i = 0; i < nums.size(); i++) {
			output[i] = beforProduct[i] * afterProduct[i];
		}
		return output;
	}
};

设置两个memo数组：beforeProduct和afterProduct，分别记录[0,i)的乘积和(i,length-1]的乘积。计算这两个数组的过程是采用动态规划。之后填充output输出数组即可。

法II

由于输出数组output不算在空间复杂度内，那么我们可以将beforeProduct和afterProduct数组用输出数组来计算。先把输出数组当作 beforeProduct 数组来计算，然后再动态构造 afterProduct 数组得到结果。让我们来看看基于这个思想的算法.

class Solution {
public:
	vector<int> productExceptSelf(vector<int>& nums) {
		vector<int> output(nums.size(), 1);
		for (int i = 1; i < nums.size(); i++) {//从前往后
			output[i] = output[i - 1] * nums[i - 1];
		}
		int afterProduct = nums[nums.size() - 1];
		for (int i = nums.size() - 2; i >= 0; i--) {//从后往前
			output[i] *= afterProduct;
			afterProduct *= nums[i];
		}
		return output;
	}
};