238. Product of Array Except Self

leetcode 238

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].

解题思路：可以先求得数组中所有数的乘积，然后分别除以当前的数。需要考虑的特殊情况就是，当数组元素为0时。

class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) {
		int res=1;
		bool isZero=false;
		int pos;
        for(int i=0;i<nums.size();i++)
        {
            if(!isZero&&nums[i]==0)//遇到第一个0
            {
    			isZero=true;
    			pos=i;
    			continue;
            }
    		if(isZero&&nums[i]==0)//遇到第二个0,此时输出的数组全为0
    		{
    		    vector<int> output(nums.size(),0);
    		    return output;
    		}
    		res=res*nums[i];
        }
		vector<int> output(nums.size(),0);
		if(isZero)//只有一个0时，输出数组中只有原数组0的位置为非0，其他全为0
		{
		    output[pos]=res;
		    return output;
		}
		for(int i=0;i<nums.size();i++)
			output[i]=res/nums[i];
		return output;
    }
};