1822. 数组元素积的符号

152

这是我参与11月更文挑战的第25天,活动详情查看:2021最后一次更文挑战

1822. 数组元素积的符号

已知函数 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

解题思路

因为函数 signFunc(x) 将会根据 x 的正负返回特定值,而我们需要返回的结果为 signFunc(product)(product 为数组 nums 中所有元素值的乘积),所以我们不需要关心我们最后乘积的大小,我们只需要关心乘积的正负号。因此我们只需要处理:

  1. 一旦出现为0的元素,那么整个结果必将为0,所以我们可以直接返回0,不需要进行后面的计算
  2. 一旦出现负数,将原来的符号置为相反数,如果原来为1,则转化为-1.如果原来为-1,则转换为1.
  3. 如果遇到的是正数,对当前的符号没有影响,因此不需要做任何的处理
  4. 一开始初始化的符号应该为1

代码

class Solution {
public:
    int arraySign(vector<int>& nums) {
        int sum=1;
        for (auto c:nums) {
            if (c==0) return 0;
            if (c<0) sum=-sum;
        }
        return sum;
    }
};
  • 时间复杂度O(n),n为nums数组的长度
  • 空间复杂度O(1),只需要使用一个变量表示符号