【剑指offer】数字在排序数组中出现的次数

转载请注明出处：http://blog.csdn.net/ns_code/article/details/27364557

题目描写叙述：

统计一个数字在排序数组中出现的次数。

输入：

每一个測试案例包括两行：

第一行有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

    以下贴上我依照自己思路写的代码：

#include<stdio.h>
#include<stdlib.h>

/*
二分查找法计算key出现的次数
*/
int CountTimesInArrays(int *arr,int len,int key)
{
	if(arr==NULL || len<1)
		return 0;

	int start = 0;
	int end = len-1;
	int mid;
	while(start <= end)
	{
		mid = (start+end)>>1;
		if(arr[mid] == key)
			break;
		else if(arr[mid] > key)
			end = mid-1;
		else 
			start = mid+1;
	}


	//包括了出现0次的情况
	int times = 0;
	if(start <= end)
	{
		int i;
		times = 1;
		for(i=mid+1;i<=end;i++)
			if(arr[i] == key)
				times++;
		for(i=mid-1;i>=start;i--)
			if(arr[i] == key)
				times++;
	}
	return times;
} 

int main()
{
	int n;
	while(scanf("%d",&n) != EOF)
	{
		int *arr = (int *)malloc(n*sizeof(int));
		if(arr == NULL)
			exit(EXIT_FAILURE);

		int i;
		for(i=0;i<n;i++)
			scanf("%d",arr+i);
		int m;
		scanf("%d",&m);
		for(i=0;i<m;i++)
		{
			int k;
			scanf("%d",&k);
			printf("%d\n",CountTimesInArrays(arr,n,k));
		}

		free(arr);
		arr = NULL;
	}
	return 0;
}