[LeedCode OJ]#238 Product of Array Except Self

【 声明:版权所有,转载请标明出处,请勿用于商业用途。  联系信箱:[email protected]

题目链接:https://leetcode.com/problems/product-of-array-except-self/

题意:
给出一个数组nums,根据这个数组得到另外一个数组ans,使得ans[i]是所有nums内除了nums[i]的乘积

思路:
首先我们对于0要特殊考虑
当没有0的时候,那么只需要算出总乘积,对于当前求的位只需要用总乘积除以当前位的值就能得出新数组该位的值
当只有1个0的时候,那么只有0位的值是其他所有位的乘积,其他位都是0
当有两个以上的0的时候,则新数组全是0

class Solution
{
public:
    vector<int> productExceptSelf(vector<int>& nums)
    {
        int len = nums.size();
        int i,j;
        vector<int> ans;
        int s = 1,cnt = 0;
        for(i = 0; i<len; i++)
        {
            if(nums[i])
                s*=nums[i];
            else
                cnt++;
        }
        for(i = 0; i<len; i++)
        {
            if(cnt==0)
            {
                ans.push_back(s/nums[i]);
            }
            else if(cnt==1)
            {
                if(nums[i])
                    ans.push_back(0);
                else
                    ans.push_back(s);
            }
            else
            {
                ans.push_back(0);
            }
        }
        return ans;
    }
};


你可能感兴趣的:(leedcode)