LeetCode题目笔记——1822. 数组元素积的符号 C++一次遍历

题目描述

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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/sign-of-the-product-of-an-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题目难度——简单

方法一：模拟

  既然题目说要返回数组nums中所有元素乘积的符号，那么我们就可以按照要求，遍历数组，做n次乘法，再看结果的符号来返回。

代码/C++

class Solution {
public:
    int arraySign(vector<int>& nums) {
        long res = 1;
        for (auto &n: nums){
            res *= n;
        }
        if (res == 0){
      		return 0;
      	}
      	else if(res > 0){
      		return 1;
      	}
      	else{
      		return -1;
      	}
    }
};

方法二：看负数的数量

  一个一个的乘，这不是欺负老实人嘛，虽然提示说数组最长也不过1000，但做1000次乘法也还是挺累人的（CPU:你说啥？）。最终结果的符号取决于数组中负数的数量，上过小学的我们都知道，奇数次那结果就是负的，偶数次那结果就是正的，所以我们只需要数一下有多少负数就行。如果中间碰到了0，直接返回0就完事儿。

代码

class Solution {
public:
    int arraySign(vector<int>& nums) {
        int tag = 1;
        for (auto &n: nums){
            if(n < 0){
                tag = -tag;
            }
            else if(n == 0){
                return 0;
            }
        }
        return tag;
    }
};

总结

  两种方法都是O(N)，第一种由于做了每次循环都做了乘法，所以实际上是要慢一点的，空间是O(1)，只用到了一两个变量。