238. 除自身以外数组的乘积

  1. 除自身以外数组的乘积
    给你一个整数数组 nums,返回 数组 answer ,其中 answer[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积 。

题目数据 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内。

请不要使用除法,且在 O(n) 时间复杂度内完成此题。

示例 1:

输入: nums = [1,2,3,4]
输出: [24,12,8,6]
示例 2:

输入: nums = [-1,1,0,-3,3]
输出: [0,0,9,0,0]

class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) {
        vector<int> result;
        int mulRet = 1, nonZeroRet = 1;
        int sumZero = 0;
        for(int i = 0; i < nums.size(); i++) {
            mulRet *= nums[i];
            if(nums[i] != 0) {
                nonZeroRet *= nums[i];
            } else {
                sumZero++;
            }
        }

        for(int i = 0; i < nums.size(); i++) {
            if(nums[i] == 0) {
                if(sumZero > 1) {
                    result.push_back(0);
                } else {
                    result.push_back(nonZeroRet);
                }
            } else {
                if(sumZero > 0) {
                    result.push_back(0);
                } else {
                    result.push_back(nonZeroRet / nums[i]);
                }
            }
        }
        return result;
    }
};

你可能感兴趣的:(leetcode算法刷题记录,算法,leetcode,数据结构)