小米2013校园招聘笔试题的第一道程序设计题

1、数组乘积（15分）
输入：一个长度为n的整数数组input
输出：一个长度为n的整数数组result，满足result[i] = input数组中除了input[i]之外所有数的乘积（假设不会溢出）。比如输入：input = {2,3,4,5}，输出result = {60,40,30,24}
程序时间和空间复杂度越小越好。
C/C++：
int *cal(int* input , int n);

Java:

int[] cal(int[] input);

这道题看起来很简单，许多人看完题目就会想到将所有的数相乘，得出总的乘机，然后再用总的乘机去分别处以input数组中的数就是要求的结果。

我们可以很容易的写出下面的代码:

int *cal(int *input,int n)
{
	assert(input!=NULL && n>0);//注意判断传入参数的合法性
	int i=0;
	int prduct = 1;
	int *result = new int[n];
	if(result == NULL)  //分配内存失败，不要忘记释放申请的堆内存
		exit(0);
	for(i=0; i<n; i++)
		prduct *= input[i];
	for(i=0; i<n; i++)
		result[i] = prduct/input[i];
	return result;
}

在测试时如果我们用的用例数组是这样的{0,2,3,4}，即数组中有1个或多个0时，程序就会在运行时崩溃，应为我们得到的prduct的值是0.

下面是参考Hackbuteer1的这篇博客http://blog.csdn.net/hackbuteer1/article/details/8484974中给出的代码，代码略有改动，增加了程序的健壮性

int *cal(int *input,int n)
{
	assert(input!=NULL && n>0);//注意判断传入参数的合法性	
	int i;
	int *result = new int[n];
	if(result == NULL)  //分配内存失败，不要忘记释放申请的堆内存
		exit(0);
	result[0] = 1;
	for(i=1; i<n; i++)
		result[i] = result[i-1]*input[i-1];
	result[0] = input[n-1];
	for(i=n-2; i>0; i--)
	{
		result[i] *= result[0];
		result[0] *= input[i];
	}
	return result;
}

一道小小的编程题，看似简单，其实也不简单。有时笔试面试时所出的编程题，出的题并不是很难，遇到这样的题时要小心的是对程序输入参数的检查和问题考虑的是否全面！