面试题66-构建乘积数组

题目

给定一个数组 A[0,1,…,n-1]，请构建一个数组 B[0,1,…,n-1]，其中 B[i] 的值是数组 A 中除了下标 i 以外的元素的积, 即 B[i]=A[0]×A[1]×…×A[i-1]×A[i+1]×…×A[n-1]。不能使用除法。

解题思路

本题的难点在于 不能使用除法 ，即需要 只用乘法 生成数组 B 。根据题目对 B[i] 的定义，可列表格，如下图所示。

根据表格的主对角线（全为 1 ），可将表格分为 上三角 和 下三角 两部分。分别迭代计算下三角和上三角两部分的乘积，即可 不使用除法 就获得结果。

C++实现

class Solution {
public:
	vector<int> constructArr(vector<int>& a) {
		int n = a.size();
		if (n < 1)
			return vector<int>();
		vector<int>b(n), c(n), d(n);

		c[0] = 1; d[n - 1] = 1;
		for (int i = 1; i < n; ++i)
		{
			c[i] = c[i - 1] * a[i - 1];//下三角
		}
		for (int i = n - 2; i >= 0; --i)
		{
			d[i] = d[i + 1] * a[i + 1];//上三角
			b[i] = c[i] * d[i]; 
		}
		b[n - 1] = c[n - 1] * d[n - 1];  //n-1索引的元素补充
		return b;
	}
};

1. 时间复杂度：O（N）其中 N 为数组长度，两轮遍历数组 a ，使用 O(N) 时间。
2. 空间复杂度：O（N）