【劍指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;
}