【剑指】66.构建乘积数组

题目描述

给定一个数组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]。不能使用除法。

算法分析

B[i]的值可以看作下图的矩阵中每行的乘积。

下三角用连乘可以很容求得，上三角，从下向上也是连乘。
因此，先计算下三角中的连乘，即我们先算出B[i]中的一部分，然后倒过来按上三角中的分布规律，把另一部分也乘进去。

提交代码：

class Solution {
public:
	vector<int> multiply(const vector<int>& A) {
		if (A.empty())
			return vector<int>();

		vector<int> B(A .size(), 1);

		for (int i = 1; i < A.size(); ++i)
			B[i] = B[i - 1] * A[i - 1];

		int temp = 1;
		for (int i = A.size() - 2; i >= 0; --i)
		{
			temp *= A[i + 1];
			B[i] *= temp;
		}

		return B;
	}
};

测试代码：

//================= Test Code =================
static void test(char* testName, const vector<int>& input, const vector<int>& expected)
{
	printf("%s Begins: ", testName);
	Solution s;
	vector<int> output = s.multiply(input);
	if (output == expected)
		printf("Passed.\n");
	else
		printf("FAILED.\n");
}

static void test1()
{
	// 输入数组中没有0
	int input[] = { 1, 2, 3, 4, 5 };
	int expected[] = { 120, 60, 40, 30, 24 };

	test("Test1", vector<int>(input, input + sizeof(input) / sizeof(int)),
		vector<int>(expected, expected + sizeof(expected) / sizeof(int)));
}

static void test2()
{
	// 输入数组中有一个0
	int input[] = { 1, 2, 0, 4, 5 };
	int expected[] = { 0, 0, 40, 0, 0 };

	test("Test2", vector<int>(input, input + sizeof(input) / sizeof(int)),
		vector<int>(expected, expected + sizeof(expected) / sizeof(int)));
}

static void test3()
{
	// 输入数组中有两个0
	int input[] = { 1, 2, 0, 4, 0 };
	int expected[] = { 0, 0, 0, 0, 0 };

	test("Test3", vector<int>(input, input + sizeof(input) / sizeof(int)),
		vector<int>(expected, expected + sizeof(expected) / sizeof(int)));
}

static void test4()
{
	// 输入数组中有正、负数
	int input[] = { 1, -2, 3, -4, 5 };
	int expected[] = { 120, -60, 40, -30, 24 };

	test("Test4", vector<int>(input, input + sizeof(input) / sizeof(int)),
		vector<int>(expected, expected + sizeof(expected) / sizeof(int)));
}

static void test5()
{
	// 输入输入中只有两个数字
	int input[] = { 1, -2 };
	int expected[] = { -2, 1 };

	test("Test5", vector<int>(input, input + sizeof(input) / sizeof(int)),
		vector<int>(expected, expected + sizeof(expected) / sizeof(int)));
}

int main(int argc, char* argv[])
{
	test1();
	test2();
	test3();
	test4();
	test5();

	return 0;
}