题目描述

给定长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。

示例:

输入: [1,2,3,4]
输出: [24,12,8,6]
说明: 请不要使用除法,且在 O(n) 时间复杂度内完成此题。

进阶:

你可以在常数空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)

算法

题中明确要求不能使用除法,本可以使用O(n^2)的双for暴力循环求解,但又要求在 O(n) 时间复杂度内。这就需要开动下脑筋,先来分析一下输出是怎么的得到:

24 = 2 * 3* 4
12 = 1 * 3 4
8 = 1
2 * 4
6 = 1 * 2 *3

记保存输出的向量为vec,长度为length。

显然vec[i]上需要填的值 = 前i个数累乘 * 后(length – i – 1)个数累乘

仔细一思考,这个好像可以和动态规划有些关联,毕竟涉及到前i个、后几个连乘这样的关系。现在还处在探索问题的阶段,将问题分为两部分思考,一个是从前往后走,一个是从后往前走,保存它们的累乘结果到2个数组中(先不考虑进阶部分要求的常数空间复杂度),观察这2个数组和我们最终的输出有什么关联。

从前往后

nums 1 2 3 4
dp_forward 1 1 2 6
if i > 0:
    dp[i] = (nums[0]到nums[i-1]的累乘)
else:
    dp[0] = 1
"""
dp数组的前后值有这么一个数学关系:
                dp[i] = dp[i-1] * nums[i-1]
"""

从后往前

为了方便阅读与理解,将nums反转为{4,3,2,1}。套用上面得到的数学关系,可以很轻松的到下面的结果:

nums 4 3 2 1
dp_backward 1 4 12 24

将nums反转为正常顺序{1,2,3,4},表格变为:

nums 1 2 3 4
dp_backward 24 12 4 1

最后一步

我们已经得到从前向后累乘与从后向前累乘的各个位置上的结果,将它们放在一起:

nums 1 2 3 4
dp_forward 1 1 2 6
dp_backward 24 12 4 1

答案呼之欲出,将dp_forward和dp_backward一行对应相乘得到{24,12,8,6}正是需要的输出。

我敢保证这不是偶然,让我们在分析一遍下面这个式子:
vec[i] = 前i个数累乘 * 后(length – i – 1)个数累乘

前i个数累乘正好保存在dp_forward[i]中;同理,后(length – i – 1)个数累乘正好保存在dp_backward[i]中。正因为这样,所以对应相乘,才正好是需要的答案。

进阶部分

在常数空间复杂度内完成这个题目。 出于对空间复杂度分析的目的,输出数组不被视为额外空间。

也就是说,我们最多可以使用一个大小为length的数组外加几个常数这么多的存储空间。

我的想法是从前往后这部分一定要保存在输出数组中,后面从后往前这部分可以直接更新在输出数组中。具体不在赘述,看代码里的注释更好理解。

代码

class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) {
        int length = nums.size();
        
        // 输出数组
        vector<int> vec(length, 1);
        
        /*** 两次遍历,从前往后一次,从后往前一次 ***/
        
        // 从前往后
        int mul = 1;
        
        for (int i = 1; i < length; i++)
        {
            mul *= nums[i-1];
            vec[i] = mul;
        }
        
        // 从后往前
        mul = 1;
        for (int i = length - 2; i >= 0; i--)
        {
            mul *= nums[i+1];
            // 更新在输出数组上,觉得抽象可以动笔画一画
            vec[i] *= mul;
        }
        
        return vec;
    }
};

版权声明:本文为shayue原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://www.cnblogs.com/shayue/p/10395307.html