LeetCode 238: Product of Array Except Self

LeetCode 238: Product of Array Except Self

题目描述

给定一个数组 A[0, 1,..., n-1]，请构建一个数组 B[0, 1,..., n-1]，其中 B 中的元素 B[i]=A[0]*A[1]*...*A[i-1]*A[i+1]*...*A[n-1]。要求不能使用除法。

解题思路

在O(N)时间复杂度下，分别进行自左向右、自右向左两次累乘即可。

AC代码

class Solution {
public:
	vector productExceptSelf(vector& nums) {
		int n = nums.size();
		if (n == 0)
			return vector();
		vector res(n);
		//自左向右累乘
		for (int i = 0, product = 1; i < n; product *= nums[i], ++i)
			res[i] = product;
		//自右向左累乘
		for (int i = n-1, product = 1; i >= 0; product *= nums[i], --i)
			res[i] *= product;
		return res;
	}
};