题目1349:数字在排序数组中出现的次数
时间限制:1 秒内存限制:32 兆特殊判题:否提交:2039解决:608
题目描述:
统计一个数字在排序数组中出现的次数。
输入:
每个测试案例包括两行:
第一行有1个整数n,表示数组的大小。1<=n <= 10^6。
第二行有n个整数,表示数组元素,每个元素均为int。
第三行有1个整数m,表示接下来有m次查询。1<=m<=10^3。
下面有m行,每行有一个整数k,表示要查询的数。
输出:
对应每个测试案例,有m行输出,每行1整数,表示数组中该数字出现的次数。
样例输入:
8
1 2 3 3 3 3 4 5
1
3
样例输出:
4
下面代码没有完全通过oj,暂时没找到原因。oj地址
#include <iostream>
#include<stdio.h>
using namespace std;
int getFirstK(int* data,int length,int k,int start,int end){
if(start>end){
return -1;
}
int middleIndex = (start+end)/2;
int middleData = data[middleIndex];
if(middleData==k){
if((middleIndex>0&&data[middleIndex-1]!=k)||middleIndex==0){
return middleIndex;
}else{
end = middleIndex-1;
}
}else if(middleIndex>k){
end = middleIndex-1;
}else{
start = middleIndex+1;
}
return getFirstK(data,length,k,start,end);
}
int getLastK(int* data,int length,int k,int start,int end){
if(start>end){
return -1;
}
int middleIndex = (start+end)/2;
int middleData = data[middleIndex];
if(middleData==k){
if((middleIndex<length-1&&data[middleIndex+1]!=k)||middleIndex==length-1){
return middleIndex;
}else{
start = middleIndex+1;
}
}else if(middleIndex<k){
start = middleIndex+1;
}else{
end = middleIndex-1;
}
return getLastK(data,length,k,start,end);
}
int getNumberOfK(int* data,int length,int k){
int number = 0;
if(data!=NULL&&length>0){
int first = getFirstK(data,length,k,0,length-1);
int last = getLastK(data,length,k,0,length-1);
if(first>-1&&last>-1){
number = last-first+1;
}
}
return number;
}
int main()
{
int n,m;
while(scanf("%d",&n)!=EOF){
int* data = new int[n];
for(int i=0;i<n;i++){
scanf("%d",&data[i]);
}
scanf("%d",&m);
while(m--){
int k ;
scanf("%d",&k);
printf("%d\n",getNumberOfK(data,n,k));
}
}
return 0;
}