数组乘积--数组中除第i个外的所有其他数组元素的乘积
Google曾经的一道面试题,腾讯的实习笔试也出了这道题,小米笔试也出了这道题
要想降低时间复杂度,就要拿空间还换
要想降低时间复杂度,就要拿空间还换
数组乘积(15分)
输入:一个长度为n的整数数组input
输出:一个长度为n的整数数组result,满足result[i] = input数组中除了input[i]之外所有数的乘积(假设不会溢出)。比如输入:input = {2,3,4,5},输出result = {60,40,30,24}
程序时间和空间复杂度越小越好。
保存i位置前的乘积到result[i],再用数组保存i位置后的乘积,结果相乘,即可。
数组下标 |
0 |
1 |
2 |
3 |
4 |
source数组 |
1 |
2 |
3 |
4 |
5 |
Prefix数组 |
1 |
1 |
2 |
6 |
24 |
Suffix数组 |
120 |
60 |
20 |
5 |
1 |
Result数组 |
120 |
60 |
40 |
30 |
20 |
prefix[i]代表从前往后到source[i]之前的累乘:即prefix[i] = source[0]*source[1]*....source[i-2]*source[i-1] =prefix[i-1]*source[i-1];而prefix[0],没有用,正好设为1.
suffix[n-i]代表从后往前到source[i]之前的累乘:即suffix[n-i] = source[n-1]*source[n-2]*....*source[n-i+1] =suffix[n-i+1]*source[n-i+1];;
suffix[n-1]没有用,正好设为1.
从而 result[i] = prefix[i]*suffix[i];
#include <iostream> using namespace std; int main() { int n; cin>>n; int *result = new int[n]; int *prefix = new int[n]; int *suffix = new int[n]; for (int i=0;i<n;i++) { cin>>result[i]; } prefix[0] = 1; for (int i=1;i<n;i++) //从前往后累积 { prefix[i] = prefix[i-1] * result[i-1]; } suffix[n-1] = 1; for (int j =2;j<=n;j++) { suffix[n-j] = suffix[n-j+1] * result[n-j+1]; //从后往前累积 } for (int k =0;k<n;k++) { result[k]= prefix[k] * suffix[k]; } for (int k =0;k<n;k++) cout<<result[k]<<endl; return 0;
具体见链接:
http://my.oschina.net/liuzeli/blog/128062
只用一个数组的方法如下:
保存i位置前的乘积到result[i],再用变量保存i位置后的乘积,结果相乘,即可。(注意与开两个数组的区别就在于此)
即:从后往前的时候设置一个临时变量q,不用再保存从后往前的累积数,而是一算出累积数就乘以从前往后的累积数result[i],也即是边算边存(
1 /* 2 * 一个长度为n的整数数组result,满足result[i]=除input[i]之外所有数的乘积(不溢出),比如 3 * 输入input={2,3,4,5};输出 result={60,40,30,24}; 4 */ 5 /* 6 * 方法一:判断有0的情况,如果有0则其他都为0.如果没0,可使用先求全部乘积,再除以自身。 7 * 方法二:先保存i位置前的乘积到result[i],再用一变量保存i位置后的乘积,结果相乘,即可。 8 */ 9 #include <stdio.h> 10 //#include <stdlib.h> 11 void pr_arr(int arr[],int n) 12 { 13 int i; 14 for( i=0;i<n;i++) 15 { 16 printf("%d ",arr[i]); 17 } 18 printf(" "); 19 } 20 21 void arrayMultiply(int input[],int result[],int n)//方法二 22 { 23 if(n<=0) return; 24 result[0]=1; 25 int i; 26 for( i =1;i<n;i++) //从1位置开始,result[i]表示input i位置前的乘积 27 { 28 result[i]=result[i-1]*input[i-1]; 29 } 30 int q=1; 31 for( i=n-2;i>=0;--i) //从倒数第二个开始,q表示input i位置后的乘积 32 { 33 q*=input[i+1]; 34 result[i]*=q; 35 } 36 pr_arr(result,n); 37 } 38 int main(void) { 39 int s[]={2,3,4,5,7,8,6}; 40 int n=sizeof(s)/sizeof(int); 41 42 int *result = new int[n]; 43 arrayMultiply(s,result,n); 44 return 0; 45 }

其中小米2013年校园招聘出了类似的题:
数组乘积(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); |
测试代码:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
#include<iostream>
using namespace std;
int *cal( int *input
, int n)
{
int i;
int *result
= new int [n];
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;
for ( int i
= 0 ; i < n ; ++i)
cout<<result[i]<< "," ;
cout<<endl;
return 0;
}
int main()
{
int input[]
= {2 , 3 , 4 , 5};
int length
= sizeof (input)
/ sizeof ( int );
*cal(input
, length);
return 0;
}
|
声明:该文观点仅代表作者本人,牛骨文系教育信息发布平台,牛骨文仅提供信息存储空间服务。