题目描述:
给定长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
示例:
输入: [1,2,3,4]
输出: [24,12,8,6]
说明: 请不要使用除法,且在 O(n) 时间复杂度内完成此题。
进阶:
你可以在常数空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/product-of-array-except-self
解题思路:
1、“不能使用除法”意味着不能使用最简单的两次遍历去解这个题,即第一次循环所有数累乘,第二次循环时每次除以相应位置的数据,求出返回数组。有一个好处是无形之中减少了一个坑,即如果有一个数是0的情况;
2、“O(n) 时间复杂度”排除了最基础的暴力解法,即两个for循环嵌套的形式;
3、需要知道除了nums[n]之外的所有数的乘积,即以nums[n]为分界线,之前的数据求出一个累乘结果,之后的数据求出一个累乘结果,两个数据相乘即可;
4、对于nums[n]之后的数据累乘,我做了一次逆序遍历,将后 numsSize-(n+1) 个数据的累乘结果,存到数组return_array[n+1]中,即 return_array[n] 中存的数据是 nums[n] * nums[n+1] * ...* nums[ numsSize -1 ];
(我在这里直接用了返回数组来存这个逆序遍历的数据,因为每个下标对应的数据都只会用一次,所以直接数组复用,减少需要开辟空间的大小)
5、对于nums[n]之前的数据累乘,在顺序遍历的同时直接做就好了;
留下的疑问:
1、定义了一个顺序遍历时累乘用到的int变量,返回数组是返回大小的值+1,也就是说,除了返回数组之外,不论输入数组长度为多少,用到的字节数都为两个int的字节大小,算不算常数空间复杂度?
int* productExceptSelf(int* nums, int numsSize, int* returnSize)
{
int *return_array = (int *)malloc(sizeof(int)*(numsSize + 1));
int mul_res = 1;
return_array[numsSize] = 1; //降低for循环中条件数量,满足式子一致性
for (int i = numsSize - 1; i > 0; i--)
{
return_array[i] = return_array[i + 1] * nums[i];
}
for (int i = 0; i < numsSize; i++)
{
return_array[i] = mul_res * return_array[i + 1];
mul_res *= nums[i];
}
*returnSize = numsSize;
return return_array;
}