【每日一题】LeetCode. 238. 除自身以外数组的乘

每日一题，防止痴呆 = =

一、题目大意

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/product-of-array-except-self

二、题目思路以及AC代码

好像偶尔有几天没有做题，这习惯不太好，以后要坚持！！！

思路：前缀数组、后缀数组

这道题其实思路很明显，要求取除该元素以外所有元素的乘积，无非就是求该元素前面元素的乘积乘该元素后面元素的乘积嘛，这样我们只要维护前缀数组和后缀数组就可以在O(n)复杂度内进行计算了。

但题目后来要求我们只用常数空间复杂度，而且括号里说输出数据的空间不算，那很自然想到利用输出的这部分空间计算其中的一个数组（前缀or后缀），然后另一个数组可以利用一个变量动态生成，也可以利用nums去存储，反正占用的空间都是常数级的。

AC代码

空间复杂度为O(n)，时间复杂度为O(n)

class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) {
        vector<int> res;

        int n_size = nums.size();
        if (!n_size) return res;

        int* pre = new int[n_size];
        int* pro = new int[n_size];

        pre[0] = 1;
        for (int i=1;i<n_size;i++) {
            pre[i] = pre[i-1] * nums[i-1];
        }

        pro[n_size-1] = 1;
        for (int i=n_size-2;i>=0;i--) {
            pro[i] = pro[i+1] * nums[i+1];
        }

        for (int i=0;i<n_size;i++) {
            res.push_back(pre[i] * pro[i]);
        }

        return res;
    }
};

空间复杂度为O(1)，时间复杂度为O(n)

class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) {
        int n_size = nums.size();

        vector<int> res(n_size);
        if (!n_size) return res;

        res[n_size-1] = 1;
        for (int i=n_size-2;i>=0;i--) {
            res[i] = res[i+1] * nums[i+1];
        }

        int last = nums[0];
        nums[0] = 1;
        for (int i=1;i<n_size;i++) {
            int tmp = last * nums[i-1];
            last = nums[i];
            nums[i] = tmp;
        }

        for (int i=0;i<n_size;i++) {
            res[i] *= nums[i];
        }

        return res;
    }
};

如果有问题，欢迎大家指正！！！