程序员面试的那些小事数据结构和算法分析程序员

lintcode 数组剔除元素后的乘积

2016-12-29  本文已影响28人  yzawyx0220

给定一个整数数组A。
定义B[i] = A[0] * ... * A[i-1] * A[i+1] * ... * A[n-1], 计算B的时候请不要使用除法。
样例
给出A=[1, 2, 3],返回 B为[6, 3, 2]。

B[i]等于A[i]左边的数的乘积乘以A[i]右边的数的乘积,只需要知道这个这道题就非常简单了。

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> res(nums.size(),1);
        long long temp = 1;
        for (int i = 0;i < nums.size();i++) {
            res[i] = temp;
            temp *= nums[i];
        }
        temp = 1;
        for (int i = nums.size()-1;i >= 0;i--) {
            res[i] *= temp;
            temp *= nums[i];
        }
        return res;
    }
};
上一篇下一篇

猜你喜欢

热点阅读