Lintcode 数组剔除元素后的乘积

最新推荐文章于 2023-04-22 20:06:13 发布

原创最新推荐文章于 2023-04-22 20:06:13 发布 · 658 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#lintcode #程序员 #算法 #c++ #面试

LintCode编程专栏收录该内容

58 篇文章

订阅专栏

本文介绍了一种高效算法，用于计算一个整数数组中每个元素对应的排除自身后的乘积结果数组，且整个过程不使用除法操作。通过前后遍历数组并累积乘积的方式，最终得到目标结果。

给定一个整数数组A。

定义B[i] = A[0] * ... * A[i-1] * A[i+1] * ... * A[n-1]，计算B的时候请不要使用除法。

您在真实的面试中是否遇到过这个题？

Yes

样例

给出A=[1, 2, 3]，返回 B为[6, 3, 2]

标签

class Solution {
public:
    /**
     * @param A: Given an integers array A
     * @return: A long long array B and B[i]= A[0] * ... * A[i-1] * A[i+1] * ... * A[n-1]
     */
    vector<long long> productExcludeItself(vector<int> &nums) {
        // write your code here
        vector<long long> result;
        vector<long long> presult,epresult;
        long long pmul=1,epmul=1;
        int n=nums.size();
        if(1==n){
        result.push_back(1);
        return result;
        }
        for(int i=0;i<n;i++){
            pmul*=nums[i];
            presult.push_back(pmul);
        }
        for(int i=n-1;i>=0;i--){
            epmul*=nums[i];
            epresult.push_back(epmul);
        }
        
        for(int i=0;i<n;i++){
            if(0==i)
            result.push_back(epresult[n-2]);
            else if(n-1==i)
            result.push_back(presult[n-2]);
            else
            result.push_back((presult[i-1]*epresult[n-i-2]));
        }
        return result;
    }
};