1. 程式人生 > >leetcode_238. Product of Array Except Self_思維

leetcode_238. Product of Array Except Self_思維

https://leetcode.com/problems/product-of-array-except-self/

給一個vector<int> nums,輸出一個vector<int> res,res[i]為nums中除去nums[i]以外所有數的乘積。且不能使用除法運算,時間複雜度為O(n),空間複雜度儘量小。

思路:首先從左往右遍歷,對任意i,可以求得nums[0,i-1]的乘積;再從右往左遍歷,對任意i,可以求得nums[i+1,n-1]的乘積。根據題意,只需要額外需要O(1)的空間複雜度。

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