二分算法：练习题目4

题目：

给你一个按 非递减顺序 排列的数组 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)) 的算法解决此问题吗？

---

参考答案：

class Solution:
    def maximumCount(self, nums: List[int]) -> int:
        def lower_bound(nums,target):
            left = 0
            right = len(nums)-1
            while left <= right:
                mid = (left + right)//2
                if nums[mid] < target:
                    left = mid + 1
                else:
                    right = mid - 1
            return left
        pos = lower_bound(nums,1)
        neg = lower_bound(nums,0)-1
        return max(neg + 1,len(nums) - pos)

注意，pos最后指定位置为正整数开始位置，neg为负整数结束位置。正整数开始位置即>=1的位置，负整数结束位置即>=0的位置-1。这样就都变为>=的判断情况，又可以用：二分算法：练习题目1方法求解