[LeetCode] 238 - Product of Array Except Self

Given an array of n integers where n > 1, nums, return an array output such that output[i] is equal to the product of all the elements of nums except nums[i].

Solve it without division and in O(n).

For example, given [1,2,3,4], return [24,12,8,6].

 

class Solution {
public:
  vector<int> productExceptSelf(vector<int>& nums) {
    int size = nums.size();
    vector<int> res(size, 1);
    int cur = 1;
    for (int i = 1; i < size; ++i) {
      cur *= nums[i - 1];
      res[i] *= nums[i];
    }
    cur = 1;
    for (int i = size -1; i >= 0; --i) {
      res[i] *= cur;
      cur *= nums[i];
    }
    return res;
  }
};

posted on 2015-08-28 11:44  tuituji  阅读(132)  评论(0编辑  收藏  举报

导航