2529.正整数和负整数的最大计数-CSDN博客

本文链接：https://blog.csdn.net/qq_43022896/article/details/137528788

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

思路

直接遍历，分别计算正整数和负整数的个数，取最大值

注意到数组有序的特点，使用二分查找，分别计算插入0的下标位置和插入1的下标位置，0插入下标左边的数都为负数，1插入下标右边的数都为正数，取最大值即可。

class Solution {
    public int maximumCount(int[] nums) {
        return Math.max(insertIdx(nums, 0), nums.length - insertIdx(nums, 1));
    }

	// 个人习惯左闭右闭区间，关于二分查找各种区间以及返回值的写法可以参考leetcode灵神的题解
    public int insertIdx(int[] nums, int target) {
        int l = 0, r = nums.length - 1, mid;
        while (l <= r) {
            mid = (r - l) / 2 + l;
            if (nums[mid] >= target) r = mid - 1;
            else l = mid + 1;
        }
        return l;
    }
}