Lintcode 数组剔除元素后的乘积

最新推荐文章于 2018-02-25 22:41:17 发布

长风o

最新推荐文章于 2018-02-25 22:41:17 发布

阅读量590

点赞数 1

分类专栏： LintCode编程文章标签： lintcode 程序员算法 c++ 面试

本文链接：https://blog.csdn.net/Hk_john/article/details/68931870

版权

LintCode编程专栏收录该内容

58 篇文章 0 订阅

订阅专栏

给定一个整数数组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;
    }
};