Question

1822. 数组元素积的符号

English Version

题目描述

已知函数 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$，负数返回 $-1$， 等于 $0$ 则返回 $0$。

我们可以定义一个答案变量 ans，初始值为 $1$。

然后遍历数组每个元素 $v$，如果 $v$ 为负数，则将 ans 乘上 $-1$，如果 $v$ 为 $0$，则提前返回 $0$。

遍历结束后，返回 ans 即可。

时间复杂度 $O(n)$，其中 $n$ 为数组长度。空间复杂度 $O(1)$。

class Solution:
  def arraySign(self, nums: List[int]) -> int:
    ans = 1
    for v in nums:
      if v == 0:
        return 0
      if v < 0:
        ans *= -1
    return ans

class Solution {
  public int arraySign(int[] nums) {
    int ans = 1;
    for (int v : nums) {
      if (v == 0) {
        return 0;
      }
      if (v < 0) {
        ans *= -1;
      }
    }
    return ans;
  }
}

class Solution {
public:
  int arraySign(vector<int>& nums) {
    int ans = 1;
    for (int v : nums) {
      if (!v) return 0;
      if (v < 0) ans *= -1;
    }
    return ans;
  }
};

func arraySign(nums []int) int {
  ans := 1
  for _, v := range nums {
    if v == 0 {
      return 0
    }
    if v < 0 {
      ans *= -1
    }
  }
  return ans
}

impl Solution {
  pub fn array_sign(nums: Vec<i32>) -> i32 {
    let mut ans = 1;
    for &num in nums.iter() {
      if num == 0 {
        return 0;
      }
      if num < 0 {
        ans *= -1;
      }
    }
    ans
  }
}

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

int arraySign(int* nums, int numsSize) {
  int ans = 1;
  for (int i = 0; i < numsSize; i++) {
    if (nums[i] == 0) {
      return 0;
    }
    if (nums[i] < 0) {
      ans *= -1;
    }
  }
  return ans;
}