您的位置:首页 > 其它

【1】数字在数组中出现的次数

2014-05-23 20:09 141 查看
题目:统计一个数字k在排序数组中出现的次数。例如输入排序数组{1,2,3,3,3,3,4,5}和数字3,输出4次

方案一:扫描数组,记录第一个出现的k和最后一个k中间有多少个,时间复杂度为O(n)

方案二:由于数组是有序的,那么我们可以利用二分的思想,求出k在数组中的第一个位置和最后位置相减即可。时间复杂度为O(logN)

注意严格按照良好的C++编码风格

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

//规定没有找到返回-1
int GetFirstIndex(int *arrNum, int left, int right, int k){
if(arrNum == NULL || left > right){
return -1;
}
while(left <= right){
int mid = (left+right)>>1;
if(arrNum[mid] > k){
right = mid-1;
}
else if(arrNum[mid] < k){
left = mid+1;
}
else{
if((mid > 0) && (arrNum[mid-1] == k)){
right = mid-1;
}
else{
return mid;
}
}
}
return -1;
}

//规定没有找到返回-1
int GetLastIndex(int *arrNum, int left, int right, int k){
if(arrNum == NULL || left > right){
return -1;
}
while(left <= right){
int mid = (left+right)>>1;
if(arrNum[mid] > k){
right = mid-1;
}
else if(arrNum[mid] < k){
left = mid+1;
}
else{
if((mid < right-1) && (arrNum[mid+1] == k)){
left = mid+1;
}
else{
return mid;
}
}
}
return -1;
}

int main(){
int arrNum[] = {1,2,3,3,3,3,4,5};
//求出第一个和最后一个位置
int firstIndex = GetFirstIndex(arrNum, 0, 7, 3);
int lastIndex = GetLastIndex(arrNum, 0, 7, 3);
if(firstIndex != -1 && lastIndex != -1){
cout<<(lastIndex-firstIndex+1)<<endl;
}
else{
cout<<-1<<endl;
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: