题目描述
统计一个数字在排序数组中出现的次数。
参考思路
一、暴力法
通过一层循环遍历数组,然后统计数字出现的次数。时间复杂度O(n);
public class Solution {
public int GetNumberOfK(int [] array , int k) {
int count = 0;
for(int i=0;i<array.length;i++){
if(array[i] == k){
count++;
}
}
return count;
}
}
二、二分法
二分法的前提是有序,当这个题目出现是有序数组时,第一时间想到用二分法提高效率。首先用二分法找到目标数字,然后根据这个数字向前和向后遍历,直到遇到不相等的数为止。时间复杂度O(logN)。
public class Solution {
public int GetNumberOfK(int [] array , int k) {
int left = 0;
int right = array.length;
int index = -1;
int count = 0;
while(left < right){
int mid = (left+right)/2;
if(array[mid] == k){
count++;
index = mid;
break;
}else if(array[mid] > k){
right = mid - 1;
}
else{
left = mid + 1;
}
}
int tmpIndex = index;
if(index > 0){
while(tmpIndex >= 0){
if(array[tmpIndex--] == k){
count++;
tmpIndex--;
}
else{
break;
}
}
tmpIndex = index;
while(tmpIndex < array.length){
if(array[tmpIndex++] == k){
count++;
tmpIndex++;
}
else{
break;
}
}
}
return count;
}
}