leetcode-java-238. Product of Array Except Self

最新推荐文章于 2023-07-23 12:04:58 发布

原创最新推荐文章于 2023-07-23 12:04:58 发布 · 359 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #java

leetcode 专栏收录该内容

83 篇文章

订阅专栏

本文介绍了一种算法，用于求解给定数组中每个元素的乘积，但不包括该元素本身。通过三种情况讨论：包含两个或更多个0、仅包含一个0以及不包含0的情况，并给出具体实现。

思路：
1.当0的个数大于等于2时，返回全为0的数组
2.当0的个数为1时，0对应的返回total，其他的为0
3.当无0时，total除以当前数

public class Solution {
    public int[] productExceptSelf(int[] nums) {
        int len = nums.length,
            total = nums[0],
            flag = 0,
            i = 1;

        int[] result = new int[len];

        for(;i < len;i++) {
            if(nums[i] == 0) {
                flag++;
                if(flag >= 2) {
                    return result;
                }
            }
            if(nums[i] != 0) {
                total = total * nums[i];
            }
        }

        for(i = 0;i < len;i++) {
            if(nums[i] != 0 && flag == 1) {
                result[i] = 0;
            } else if(nums[i] != 0 && flag == 0) {
                result[i] = total/nums[i];
            } else{
                result[i] = total;
            }
        }

        return result;
    }
}