C/C++教程

LeetCode 238 Product of Array Except Self 前缀积&后缀积

本文主要是介绍LeetCode 238 Product of Array Except Self 前缀积&后缀积,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

Given an integer array nums, return an array answer such that answer[i] is equal to the product of all the elements of nums except nums[i].

The product of any prefix or suffix of nums is guaranteed to fit in a 32-bit integer.

You must write an algorithm that runs in \(O(n)\) time and without using the division operation.

Solution

当前位置的答案就是这个位置 \(pre\times suf\),我们可以迭代更新 \(pre,suf\) 不用额外的储存空间

点击查看代码
class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) {
        vector<int> ans(nums.size(),1);
        int n = nums.size();
        int pre=1, suf=1;
        for(int i=0;i<n;i++){
            ans[i]*=pre; pre*=nums[i];
        }
        for(int i=n-1;i>=0;i--){
            ans[i]*=suf;suf*=nums[i];
        }
        return ans;
    }
};
这篇关于LeetCode 238 Product of Array Except Self 前缀积&后缀积的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!