正整数和负整数的最大计数(Lc2529)——遍历计数、二分查找

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

问题简要描述：返回正整数数目和负整数数目中的最大值

解法一

Java

class Solution {
    public int maximumCount(int[] nums) {
        int a = 0, b = 0;
        for (int num : nums) {
            if (num < 0) {
                a++;
            } else if (num > 0) {
                b++;
            }
        }
        return Math.max(a, b);
    }
}

Python

class Solution:
    def maximumCount(self, nums: List[int]) -> int:
        a = sum(x < 0 for x in nums)
        b = sum(x > 0 for x in nums)
        return max(a, b)        

TypeScript

function maximumCount(nums: number[]): number {
    let a = 0, b = 0;
    for (const num of nums) {
        if (num < 0) {
            a++;
        } else if (num > 0) {
            b++;
        }
    }
    return Math.max(a, b);    
};

解法二

Java

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

    int search(int[] nums, int x) {
        int l = 0, r = nums.length;
        while (l < r) {
            int mid = (l + r) >> 1;
            if (nums[mid] >= x) {
                r = mid;
            } else {
                l = mid + 1;
            }
        }
        return l;
    }
}

 Python

class Solution:
    def maximumCount(self, nums: List[int]) -> int:
        a = bisect_left(nums, 0)
        b = len(nums) - bisect_left(nums, 1)
        return max(a, b) 

TypeScript

function maximumCount(nums: number[]): number {
    const search = (x: number) => {
        let l = 0, r = nums.length;
        while (l < r) {
            let mid = (l + r) >> 1;
            if (nums[mid] >= x) {
                r = mid;
            } else {
                l = mid + 1;
            }
        }
        return l;
    }
    const a = search(0), b = nums.length - search(1);
    return Math.max(a, b);
};