2529. 正整数和负整数的最大计数

题目

给你一个按 非递减顺序 排列的数组 nums ，返回正整数数目和负整数数目中的最大值。

• 换句话讲，如果 nums 中正整数的数目是 pos ，而负整数的数目是 neg ，返回 pos 和 neg二者中的最大值。

注意：0 既不是正整数也不是负整数。

示例 1：

输入：nums = [-2,-1,-1,1,2,3]
输出：3
解释：共有 3 个正整数和 3 个负整数。计数得到的最大值是 3 。

示例 2：

输入：nums = [-3,-2,-1,0,0,1,2]
输出：3
解释：共有 2 个正整数和 3 个负整数。计数得到的最大值是 3 。

示例 3：

输入：nums = [5,20,66,1314]
输出：4
解释：共有 4 个正整数和 0 个负整数。计数得到的最大值是 4 。

提示：

• 1 <= nums.length <= 2000
• -2000 <= nums[i] <= 2000
• nums 按 非递减顺序 排列。

进阶：

你可以设计并实现时间复杂度为 O(log(n)) 的算法解决此问题吗？

解答

这道题不用二分也能做啊。

代码

C++ 代码

class Solution {
public:
    int maximumCount(vector<int> &nums) {
        int less = lower_bound(nums.begin(), nums.end(), 0) - nums.begin();
        int great = nums.end() - upper_bound(nums.begin(), nums.end(), 0);
        
        return max(less, great);
    }
};

其中：

• lower_bound(begin, end, num) 查找第一个 >= num 的数字，找到返回该数字的迭代器，不存在则返回 end()
• upper_bound(begin, end, num) 查找第一个 > num 的数字，找到返回该数字的迭代器，不存在则返回 end()

Python 代码

class Solution:
    def maximumCount(self, nums: List[int]) -> int:
        neg_len = bisect_left(nums, 0)
        pos_len = len(nums) - bisect_right(nums, 0)

        return max(neg_len, pos_len)

其中：

• bisect_left(lst, num) 返回 lst 中 大于等于 num 的第一个下标
• bisect_right(lst, num) 返回 lst 中 大于 num 的第一个下标
• bisect(lst, num) 返回 lst 中 大于 num 的第一个下标（似乎没有这个）