Product of Array Except Self

標籤：

Given an array of n integers where n > 1, nums, return an array output such thatoutput[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].

題目分析：給定一個數組，求其第i位上除了nums[i]之外其他所有元素的乘積，組成的新的數組。
解題方法一：將原數組除0之外的所有元素相乘為total，並同時累加為0的元素，如果0的元素大於等於0，則新數組全部為0，如果只有一個元素為0，則記錄此值得位置，這個位置上的新數組元素為total，除此之外均為0.如果沒有0存在，則用total除以原位置上的值即可得新數組該位置上的值代碼如下：

vector<int> productExceptSelf(vector<int>& nums) {        int total=1;        int num=0;        int weizhi=-1;        for(int i=0;i<nums.size();i++)        {            if(nums[i]!=0)            total*=nums[i];            if(nums[i]==0)            {                num++;                weizhi=i;            }        }        vector<int> res(nums.size(),0);        if(num>1)        {             return res;        }        else if(num==1)        {           res[weizhi]=total;           return res;        }        else        {            for(int i=0;i<nums.size();i++)            res[i]=total/nums[i];            return res;        }        return res;    }

解題方法二：我們分成兩步，從前到後、從後往前。代碼如下：

 int n=nums.size();       vector<int> res(n,1);   for(int i=0;i<n;i++)   {   if(i==0) res[i]=1;   else   res[i]=res[i-1]*nums[i-1];   }   int right=1;   for(int i=n-1;i>=0;i--)   {   res[i]*=right;   right*=nums[i];   }   return res;

著作權聲明：本文為博主原創文章，未經博主允許不得轉載。

Product of Array Except Self