目录
题目:
统计一个数字在排序数组中出现的次数。
示例 1:
输入: nums = [5,7,7,8,8,10], target = 8
输出: 2
示例 2:输入: nums = [5,7,7,8,8,10], target = 6
输出: 0来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/zai-pai-xu-shu-zu-zhong-cha-zhao-shu-zi-lcof
思路:
1.这道题看起来非常容易,常规思路是遍历一遍,这样的时间复杂度是O(n),不是我们要的结果。
2.还有一种思路是,因为这道题给出的是一个排序好的数组,很容易让我们想到二分查找,因此,先二分,再向前后查找为target的元素。但是有可能n个数据有接近n个target,这样我们的时间复杂度也是O(n)。
3.本道题,我们可以找出第一个target和最后一个target,通过下标运算得到结果。我们可以通过二分查找来找到第一个target。当nums[mid]和target相等时,我们要判断前一个是不是target,是的话就向前二分,不是的话就是我们要找的first。
4.因为first和last都需要先找到一个target,为了尽可能减少时间,我们可以先找出target,然后再传入first和last。
代码:
package JianZhiOffer;
public class Day04_2 {
public static void main(String[] args) {
Solution53_1 solution53_1=new Solution53_1();
System.out.println(solution53_1.search(new int[]{1,1,1,2,2,2,3},2));
}
}
class Solution53_1 {
public int search(int[] nums, int target) {
int begin=0;
int end=nums.length-1;
int mid;
int first=0;
int last=-1;
while(begin<=end){
mid=(begin+end)/2;
if(nums[mid]==target){
first=searchFirst(nums,target,begin,end,mid);
last=searchLast(nums,target,begin,end,mid);
break;
}else if(target<nums[mid]){
end=mid-1;
}else{
begin=mid+1;
}
}
return last-first+1;
}
public int searchFirst(int[] nums,int target,int begin,int end,int mid){
while(begin<=end){
mid=(begin+end)/2;
if(nums[mid]==target){
if((mid>0&&nums[mid-1]!=target)||mid==0){
return mid;
}else{
end=mid-1;
}
}else if(target<nums[mid]){
end=mid-1;
}else{
begin=mid+1;
}
}
return -1;
}
public int searchLast(int[] nums,int target,int begin,int end,int mid){
while(begin<=end){
mid=(begin+end)/2;
if(nums[mid]==target){
if((mid<nums.length-1&&nums[mid+1]!=target)||mid==nums.length-1){
return mid;
}else{
begin=mid+1;
}
}else if(target<nums[mid]){
end=mid-1;
}else{
begin=mid+1;
}
}
return -1;
}
}