1. 程式人生 > >【LeetCode】238. 除自身以外陣列的乘積 結題報告 (C++)

【LeetCode】238. 除自身以外陣列的乘積 結題報告 (C++)

原題地址:https://leetcode-cn.com/problems/product-of-array-except-self/submissions/

題目描述:

給定長度為 n 的整數陣列 nums,其中 n > 1,返回輸出陣列 output ,其中 output[i] 等於 nums 中除 nums[i] 之外其餘各元素的乘積。

示例:

輸入: [1,2,3,4]
輸出: [24,12,8,6]
說明: 請不要使用除法,且在 O(n) 時間複雜度內完成此題。

進階:
你可以在常數空間複雜度內完成這個題目嗎?( 出於對空間複雜度分析的目的,輸出陣列不被視為額外空間。)

 

解題方案:

陣列題型,要求不能用除法,而且不能有額外的空間。

非常簡單,我們可以遍歷nums,在遍歷的過程中將對應元素累乘,例如

1  2  3  4
1  1  2  6
這樣我們就得到了對應元素左邊所有元素的乘積。然後我們反向遍歷nums,做相同操作即可。

1  2  3  4
24 12 4  1
再將兩個結果相乘即可。

1  2  3  4
24 12 8  6
這樣的話,就有額外的空間生成。其實,只需做一點調整就能解決問題:

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