283.除自身以外数组的乘积（前缀积、C解法）

努力努力的脆脆鲨2024-01-19 9:40

题目描述：

给你一个整数数组 nums，返回 数组 answer ，其中 answer[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积 。

题目数据保证数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内。

请 不要使用除法， 且在 O(n) 时间复杂度内完成此题。

示例 1:

复制代码

输入: nums = [1,2,3,4]
输出: [24,12,8,6]

示例 2:

复制代码

输入: nums = [-1,1,0,-3,3]
输出: [0,0,9,0,0]

前缀积解法：

cpp 复制代码

int* productExceptSelf(int* nums, int numsSize, int* returnSize) {
    int* ans = (int *)malloc(sizeof(int) * numsSize);
    *returnSize=numsSize;

    ans[0]=1;
    for(int i=1;i<numsSize;i++){
        ans[i]=ans[i-1]*nums[i-1];
    }
    int r=1;
    for(int i=numsSize-1;i>=0;i--){
        ans[i]*=r;
        r*=nums[i];
    }
    return ans;
}

分析：

该思路和前缀和有点类似，即计算下一个元素的积时有很大一部分计算是与计算上一个元素的积重复的，用前缀积数组存储计算值，在该值的基础上进行下一个计算，减少了重复运算，达到降低时间复杂度的效果。

每一个元素除自身以外数组的积可以分解为前缀积乘以后缀积，后缀积求法与前缀积类似，只是遍历的顺序相反。可以用两个数组，两个for循环分别求前缀积、后缀积，再用一个数组，一个for循环计算题解，这种解法思路更直接，但开了三个数组空间复杂度高。本解法选择用一个for循环将题解数组当作前缀积数组，再用一个for循环同时计算后缀积和题解，这样只开了一个数组。

需注意，函数中定义的数组（int ans[returnSize];)是局部变量，离开函数后会自动释放，不能用return传出去，必须用malloc定义数组。