LeetCode- 238. Product of Array Except Self - 思路详解 -C++

题目

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

Follow up:
Could you solve it with constant space complexity? (Note: The output array does not count as extra space for the purpose of space complexity analysis.)

翻译

给定一个数组A，具有n个元素，n大于1，返回一个数组，数组元素output[i]为，除过元素A【i】，其他元素的乘积。时间复杂度为O(n)

注：
是否可以用O(1)的空间复杂度结果问题，

思路

首先求出数组的成绩，然后对每个元素，除以它，就得到了其他元素乘积的结果，这样子，时间复杂度就是O(n)

但是这里边要注意坑，就是如果里边有0的情况。

这里我们分析得到：
1，如果成绩为0，则数组中一定有数位0.
2，如果数组有由两个以上的数据位0，则所有的结果均为0.
3，如果只有一个，则这个0处的结果不为0外，其他均为0.

代码

class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) {
        int all = 1;
        int cnt_zero = 0;
        vector<int> res;

        for(int i = 0; i < nums.size();i++){
            if(nums[i] == 0){
                cnt_zero++;
            }
            all *= nums[i];
        }
        //成绩为0，说明其中有0，如果0的个数多余1个，则所有都是0，
        //如果只有一个，则除了0元素非零外，其他元素均为0
        if(all == 0){
            if(cnt_zero > 1){
                res = vector<int>(nums.size(),0);
            }else{
                res = vector<int>(nums.size(),0);
                int pos = -1;
                for(int i = 0; i < nums.size();i++){
                    if(nums[i] == 0){
                        pos = i;
                    }
                }
                res[pos] = 1;
                for(int i = 0; i < nums.size(); i++){
                    if(i != pos){
                        res[pos] *= nums[i];
                    }
                }
            }
        }else{
            for(int i = 0; i < nums.size(); i++){
                res.push_back(all/nums[i]);
            }
        }
        return res;
    }
};