LeetCode--238

丘小羽2024-02-27 14:42

给你一个整数数组 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]

提示：

2 <= nums.length <= 105
-30 <= nums[i] <= 30
保证数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内

如果没有时间复杂度限制，这道题很简单，如下是我的解：

复制代码

class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) {
        vector<int> ans;
        for(int i=0;i<nums.size();i++)
        {
            int res=1;
            for(int j=0;j<nums.size();j++)
            {
                if(j==i)
                continue;
                res*=nums[j];
            }
            ans.push_back(res);
        }
        return ans;
    }
};

但是显然提交后显示错误，因为时间复杂度超出了题目要求的范围。

下面提供官方解答：

复制代码

class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) {
        int length = nums.size();
        vector<int> L(length,0), R(length,0);
        vector<int> answer(length);
        L[0] = 1;
        for (int i = 1; i < length; i++) {
            L[i] = nums[i - 1] * L[i - 1];
        }
        R[length - 1] = 1;
        for (int i = length - 2; i >= 0; i--) {
            R[i] = nums[i + 1] * R[i + 1];
        }
        for (int i = 0; i < length; i++) {
            answer[i] = L[i] * R[i];
        }
        return answer;
    }
};

而我抄都能抄错，注意for循环的起始条件，答案明显很简单，但是我竟然没有想到！！！可恨。