已知函数 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
思路
首先,函数使用count函数检查nums中是否存在0,如果存在0,那么就直接返回0,因为任何数乘以0都是0。
如果nums中不存在0,那么函数使用sort函数将nums排序。然后,函数使用lower_bound函数找到第一个大于等于0的元素的位置,然后减去nums.begin(),得到nums中小于0的元素的数量cnt。
最后,函数返回的cnt次方,这就是nums中所有元素的乘积的符号。因为负数的个数为偶数时,乘积为正,负数的个数为奇数时,乘积为负。
AC代码
/*
* @lc app=leetcode.cn id=1822 lang=cpp
*
* [1822] 数组元素积的符号
*/
// @lc code=start
class Solution {
public:
int arraySign(vector<int>& nums) {
if (count(nums.begin(), nums.end(), 0)) {
return 0;
}
sort(nums.begin(), nums.end());
int cnt = lower_bound(nums.begin(), nums.end(), 0) - nums.begin();
return pow(-1, cnt);
}
};
// @lc code=end