46.在排序数组中查找数字 I
统计一个数字在排序数组中出现的次数。
示例 1:
输入: nums = [5,7,7,8,8,10], target = 8
输出: 2
示例 2:
输入: nums = [5,7,7,8,8,10], target = 6
输出: 0
思路一
双指针。首尾各放一个指针,由于数组是排好序的,所以找到与target相等的那个范围即可。注意返回值,如果i和j相遇也没有找到target,那么就返回0,否则返回j-i+1。
class Solution {
public int search(int[] nums, int target) {
if(nums==null || nums.length==0 )
{
return 0;
}
int i=0;
int j=nums.length-1;
while(i<j)
{
if(nums[i]<target) i++;
if(nums[j]>target) j--;
if(nums[i]==target && nums[j]==target)
{
break;
}
}
return nums[i]==target?(j-i+1):0;
}
}
这种方法时间复杂度是O(N)。
思路二——二分
二分的精髓就是在已经排好序的数据中发现与寻找目标的联系。
数组已经排好序,如果找一个元素出现的次数,那么他们必然是相连的。当考虑相连的这段的端点时我们发现,左端点的左边都小于它,右边都大于等于它。右端点左边都小于等于它,右边都大于它。
利用这两个性质,我们可以使用两次二分,分别确定两个端点。端点找到了,用下标一减,就是次数。
class Solution {
public int search(int[] nums, int target) {
if(nums==null || nums.length==0)
{
return 0;
}
int l=0,r=nums.length-1;
while(l<r)
{
int mid=(l+r)>>1;
if(nums[mid]<target)
{
l=mid+1;
}else
{
r=mid;
}
}
if(nums[l]!=target) return 0;
int left=l;
l=0;
r=nums.length-1;
while(l<r)
{
int mid=(l+r+1)>>1;
if(nums[mid]<=target)
{
l=mid;
}else
{
r=mid-1;
}
}
return r-left+1;
}
}