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

这种做法的平均时间复杂度为O（logn）。最坏情况下为O(n),剑指offer上给的思路是两次用二分查找分别找到该数字第一次和最后一次出现的位置，这种时间复杂度平均和最坏都是O（logn），稍好些》

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

#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;
}