方法一
解题思路
最容易想到的就是直接遍历一遍数组统计个数,比较返回即可。
解题代码
class Solution {
public int maximumCount(int[] nums) {
int posNum = 0, negNum = 0;
for (int i = 0; i < nums.length; i++) {
if (nums[i] > 0) posNum++;
if (nums[i] < 0) negNum++;
}
return posNum > negNum ? posNum : negNum;
}
}
提交结果
代码优化
以上还可以优化一下,但是意义不大,时间复杂度最差还是 O(n)
。但是在一般情况下要比遍历整个数组要好一些。优化后的代码如下:
class Solution {
public int maximumCount(int[] nums) {
int zeroNum = 0, posNum = 0, negNum = 0;
if (nums[0] > 0) {
posNum = nums.length;
} else if (nums[0] == 0){
for (int i = 0; i < nums.length; i++) {
if (nums[i] == 0) zeroNum++;
if (nums[i] > 0) break;
}
posNum = nums.length - zeroNum;
} else {
for (int i = 0; i < nums.length; i++) {
if (nums[i] < 0) negNum++;
if (nums[i] == 0) zeroNum++;
if (nums[i] > 0) break;
}
posNum = nums.length - negNum - zeroNum;
}
return posNum - negNum > 0 ? posNum : negNum;
}
}
方法二
解题思路
在进阶中提到可以用时间复杂度为 O(log(n)) 的算法解决此问题,而且题目已经说明数组是按 非递减顺序 排列的。所以我们可以用 二分查找 算法进行解决。找到第一个大于等于0
的数字的下标和第一个大于等于1
的数字的下标即可。具体的二分查找算法之前的博客也写过,详情请看 LeetCode 35. 搜索插入位置(二分查找)。
解题代码
class Solution {
public int maximumCount(int[] nums) {
int negNum = Search_Bin(nums, 0);
int posNum = nums.length - Search_Bin(nums, 1);
return posNum - negNum > 0 ? posNum : negNum;
}
public int Search_Bin(int[] nums, int num) {
int l = 0, r = nums.length;
while (l < r) {
int mid = (l + r) >> 1;
if (nums[mid] >= num) r = mid;
else l = mid + 1;
}
return l;
}
}