力扣 238. 除自身以外数组的乘积

🔗 https://leetcode.cn/problems/product-of-array-except-self

题目

  • 给定一个数组,返回除了当前元素,其余所有元素的乘积
  • 不能用除法,所有数字的乘积都在 int 32 位之内

思路

  • solution 1
    • presum 的思路变成 prefix_product 和 suffix_product 的计算,当前元素的答案,就是对应的 prefix_product 和 suffix_product 的乘积
  • solution 2
    • 进阶要求仅用额外的 O(1) 空间,那遍顺序遍历一遍,逆序遍历一遍,记录对应的乘积,更新当前元素的答案

代码

cpp 复制代码
class Solution {
public:
    vector<int> solution1(vector<int>& nums) {
        int n = nums.size();
        vector<int> suffix_product(n+1), prefix_product(n+1);
        suffix_product[n] = prefix_product[0] = 1; 
        for (int i = 0; i < nums.size(); i++) {
            prefix_product[i + 1] = prefix_product[i] * nums[i];
            suffix_product[n - i -1] = suffix_product[n - i] * nums[n - i -1];
        }
        vector<int> ans(n);
        for (int i = 0; i < n; i++) {
            ans[i] = prefix_product[i] * suffix_product[i+1];
        }
        return ans; 
    }

    vector<int> solution2(vector<int>& nums) {
        int n = nums.size();
        vector<int> ans(n, 1);
        int pre_prod = 1;
        for (int i = 0; i < n; i++) {
            ans[i] = pre_prod;
            pre_prod *= nums[i];
        }
        int suf_prod = 1;
        for (int i = n - 1; i >= 0; i--) {
            ans[i] *= suf_prod;
            suf_prod *= nums[i];
        }
        return ans;
    }

    vector<int> productExceptSelf(vector<int>& nums) {
        //return solution1(nums);
        return solution2(nums);
        
    }
};
相关推荐
进击的炸酱面6 分钟前
第三章 线性模型
人工智能·算法·机器学习
立志成为大牛的小牛7 分钟前
数据结构——三十一、最小生成树(王道408)
数据结构·学习·程序人生·考研·算法
CoovallyAIHub36 分钟前
一致性模型:单步生成高质量图像,破解扩散模型速度瓶颈
深度学习·算法·计算机视觉
JMzz44 分钟前
Rust 中的数据结构选择与性能影响:从算法复杂度到硬件特性 [特殊字符]
开发语言·数据结构·后端·算法·性能优化·rust
CoovallyAIHub1 小时前
搞定边缘AI部署:开源神器RamaLama,让视觉语言模型无处不在
深度学习·算法·计算机视觉
CyberSoma1 小时前
机器人模仿学习运动基元数学编码方法还有用吗?
人工智能·算法·计算机视觉·机器人
CoovallyAIHub1 小时前
英伟达再出「神作」!黄仁勋华盛顿GTC宣布Vera Rubin超级芯片,联手诺基亚进军6G,市值直逼5万亿美元
深度学习·算法·计算机视觉
黑菜钟2 小时前
代码随想录第50天 | 图论 基础介绍(新篇章
算法·深度优先·图论
草莓熊Lotso2 小时前
《算法闯关指南:优选算法--前缀和》--27.寻找数组的中心下标,28.除自身以外数组的乘积
开发语言·c++·算法·rpc
七夜zippoe2 小时前
仓颉语言核心特性详解:类型系统与内存安全
人工智能·算法·鸿蒙·仓颉·核心实践