【LeetCode 238】

题目描述:

给一个数组,求数组的各元素乘积(不包括当前位置)。

思路:

用两个数组记录顺序数组乘积和逆序数组乘积。

代码:

class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) {
        int len = nums.size();
        vector<int> pre(len);
        vector<int> back(len);
        pre[0] = nums[0];
        for (int i=1; i<len; ++i) {
            pre[i] = pre[i-1] * nums[i];
        }
        back[len-1] = nums[len-1];
        for (int i=len-2; i>=0; --i) {
            back[i] = back[i+1] * nums[i];
        }
        vector<int> ans(len);
        for (int i=0; i<len; ++i) {
            if (i == 0) ans[i] = back[i+1];
            else if (i == len-1) ans[i] = pre[i-1];
            else ans[i] = pre[i-1] * back[i+1];
        }
        return ans;
    }
};

猜你喜欢

转载自blog.csdn.net/iCode_girl/article/details/90019779