算法--数组元素积的符号

129 阅读2分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第19天,点击查看活动详情

题目

leetcode 862. 和至少为 K 的最短子数组 难度:困难 已知函数 signFunc(x) 将会根据 x 的正负返回特定值:

如果 x 是正数,返回 1 。

如果 x 是负数,返回 -1 。

如果 x 是等于 0 ,返回 0 。

给你一个整数数组 nums 。令 product 为数组 nums 中所有元素值的乘积。

返回 signFunc(product) 。

 

示例 1:

输入:nums = [-1,-2,-3,-4,3,2,1]
输出:1
解释:数组中所有值的乘积是 144 ,且 signFunc(144) = 1

示例 2:

输入:nums = [1,5,0,2,-3]
输出:0
解释:数组中所有值的乘积是 0 ,且 signFunc(0) = 0

示例 3:

输入:nums = [-1,1,-1,1,-1]
输出:-1
解释:数组中所有值的乘积是 -1 ,且 signFunc(-1) = -1

提示:

1 <= nums.length <= 1000

-100 <= nums[i] <= 100

题解

不要陷入惯性思维了,我一开始也是,一看这么简单的题,直接遍历求出全部的乘积,然后判断返回。

其实没有必要,如果求乘积,我们还要额外处理溢出的情况。

我们直接判断当前的数字是不是负数,如果是负数就乘-1就好,不用乘实际的数,因为本身也不需要;如果遇到了0,直接返回0即可,不用继续循环;如果是正数的话不用做任何额外的处理。

/**
 * @param {number[]} nums
 * @return {number}
 */
var arraySign = function(nums) {
    let sign = 1;
    for (const num of nums) {
        if (num === 0) {
            return 0;
        }
        if (num < 0) {
            sign = -sign;
        }
    }
    return sign;
};

代码详解

如果数组中有一个元素 0,那么所有元素值的乘积肯定为 0,我们直接返回0。使用 sign 记录元素值乘积的符号,1 为表示正,-1 表示为负,初始时 sign=1。遍历整个数组,如果元素为正,那么 sign 不变,否则令 sign=−sign,最后返回 sign。