题目描述:统计数字在排序数组中出现的次数,例如输入数组{1,2,3,3,3,3,4,5,7,8}和数字3,则输出4,因为数字3在数组中出现了4次。
解:此题最简单做法是扫描数组,每当碰到与数字一样的数组元素时,次数+1。这种做法的时间复杂度为O(N),而且没有用上数组有序这个性质。
另一种做法是用二分法找到对应的数组元素下标,然后分别往数组的左边和右边扫描,找到第一个与最后一个元素。但是这种做法的时间的时间复杂度依然为O(N)。
比较好的方法是,用二分法查找出数字的起始位置和末尾位置。这样的话时间复杂度为O(logN),代码如下:
class Solution {
public:
int GetNumberOfK(vector<int> data ,int k) {
int Size = data.size();
if(Size==0) return 0;
if(data[0]>k||data[Size-1]<k) return 0;
int FirstK = GetFirstK(data,k);
int LastK = GetLastK(data,k);
if(FirstK!=-1&&LastK!=-1) return LastK-FirstK+1;
else return 0;
}
int GetFirstK(vector<int> data, int k){
int Size = data.size();
int Left = 0, Right = Size-1, IndexK = -1;
if(data[Left]==k) return Left;
int Mid;
while(Left<=Right){
Mid = (Left+Right)/2;
if(data[Mid]==k) {
if(data[Mid-1]<k) return Mid;
else Right = Mid-1;
}
else if(data[Mid]<k) Left = Mid+1;
else Right = Mid-1;
}
return -1;
}
int GetLastK(vector<int> data, int k){
int Size = data.size();
int Left = 0, Right = Size-1, IndexK = -1;
if(data[Right]==k) return Right;
int Mid;
while(Left<=Right){
Mid = (Left+Right)/2;
if(data[Mid]==k){
if(data[Mid+1]>k) return Mid;
else Left = Mid+1;
}
else if(data[Mid]>k) Right = Mid-1;
else Left = Mid+1;
}
return -1;
}
};
主函数是GetNumberOfK。函数首先要判断输入数组的有效性,如数组是否空,第一个元素是否大于K,最后一个元素是否小于K。这个判断非常重要,否则可能会导致数组溢出。之后用GetFirstK和GetLastK函数获取数组第一个K和最后一个K出现的下标。两者之差即为K出现的次数。两个函数功能类似,因此只解析GetFirstK。
函数首先判断一下第一个元素是否等于K,如果是的话可以直接返回第一个元素下标,同时也避免下方循环中算data[Mid-1]时数组溢出。
循环中不断比较中间元素和K。如果中间元素大于K,则K在数组前半段。若中间元素大于K,则K在数组前半段。如果中间元素等于K,则比较一下前一个元素是否小于K,如果是,则此元素为K第一次出现的位置,直接返回。否则元素K第一次出现的位置一定在数组的前半段。
题目来源https://www.nowcoder.com/practice/70610bf967994b22bb1c26f9ae901fa2?tpId=13&tqId=11190&tPage=1&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking