给你一个长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
来源:力扣(LeetCode)
提示:题目数据保证数组之中任意元素的全部前缀元素和后缀(甚至是整个数组)的乘积都在 32 位整数范围内。
说明: 请不要使用除法,且在 O(n) 时间复杂度内完成此题。
进阶:
你可以在常数空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
不考虑时间复杂度,首先应该是考虑两层的嵌套循环,遍历数组,在乘积数组中出去当前数,代码如下:
int* productExceptSelf(int* nums, int numsSize, int* returnSize){
int num=1;
int* output=(int*)malloc(numsSize*sizeof(int));
for(int i=0;i<numsSize;i++){
for(int j=0;j<numsSize;j++){
if(j==i)
continue;
else
num*=nums[j];
}
output[i]=num;
num=1;
}
*returnSize = numsSize;
return output;
}
但题目中对时间复杂度进行了限制,因此可将两层嵌套循环拆分成两个循环,分别用数组记录当前数左边数的乘积以及右边数的乘积,那么总的输出即为左边数的乘积乘以右边数的乘积,时间复杂度将会大大降低,代码如下:
int* productExceptSelf(int* nums, int numsSize, int* returnSize){
int* output=(int*)malloc(numsSize*sizeof(int));
int* right=(int*)malloc(numsSize*sizeof(int));
int* left=(int*)malloc(numsSize*sizeof(int));
for(int i=0;i<numsSize;i++){
if(i==0)
left[i]=1;
else
left[i]=left[i-1]*nums[i-1];
}
for(int i=numsSize-1;i>=0;i--){
if(i==numsSize-1)
right[i]=1;
else
right[i]=right[i+1]*nums[i+1];
output[i]=left[i]*right[i];
}
free(left);
free(right);
*returnSize = numsSize;
return output;
}