题目
给你一个整数数组 nums,返回 数组 answer ,其中 answer[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积 。
题目数据 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内。
请不要使用除法,且在 O(n) 时间复杂度内完成此题。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/product-of-array-except-self
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
示例
思路
数组nums之中任意元素除自身以外数组的乘积等于当前元素前缀元素和后缀元素的乘积
定义一个a数组,保存nums数组中前缀元素的乘积
再定义一个b数组,保存nums数组中后缀元素的乘积
值定义一个c数组,保存nums数组中元素前缀元素和后缀元素的乘积
所以c为返回数组
时间复杂度为O(n),但是空间复杂度不满足进阶要求,所以需要优化
居然c数组只是为了保存结果值,在保存结果值之前并没有实际作用,那么我们就可以将c数组和a数组合并,用c数组保存nums数组中前缀元素的乘积,之后直接与c数组相乘返回即可,但是还是会额外用到b数组,我们可以将b数组进行简化,变为一个变量b,每次保存
nums数组中后缀元素的乘积,这样每次遍历每次乘b即可,不需要额外空间保存他
代码
/**
* Note: The returned array must be malloced, assume caller calls free().
*/
/*
*int* productExceptSelf(int* nums, int numsSize, int* returnSize)
int* productExceptSelf:求除自身以外数组的乘积
int* nums, int numsSize:数组的长度
int* returnSize:返回数组的长度
返回值:除自身以外数组的乘积的数组
*/
int* productExceptSelf(int* nums, int numsSize, int* returnSize){
*returnSize = numsSize;
int * ans = malloc(sizeof(int) * numsSize);
int i = 0;
ans[0] = 1;
for(i = 1; i < numsSize; i++)
{
ans[i] = ans[i-1] * nums[i-1]; //计算前缀乘积
}
int R = 1;
for(i = numsSize-1; i >= 0; i--)
{
ans[i] = ans[i] * R;//保存前缀和后缀的乘积
R = R * nums[i];//计算后缀的乘积
}
return ans;
}
时间空间复杂度