leetcode_238. Product of Array Except Self_思維
阿新 • • 發佈:2018-12-08
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; } };