剑指offer 38-数字在排序数组中出现的次数

利用二分查找的思想，分别找出第一次和最后一次出现的位置，然后相减得到出现次数

#include<iostream>
using namespace std;
// 找到数组中第一个k的下标。如果数组中不存在k，返回-1
int GetFirstK(int *data, int length,int start ,int end ,int k)
{
	if(start > end)
		return -1;
	int middle = (start+end)/2;
	if(data[middle]==k)
	{
		if((middle>0 && data[middle-1]!=k) || middle==0)
			return middle;
		else
			end = middle-1;
	}
	else if(data[middle]>k)
		end = middle-1;
	else
		start = middle+1;

	return GetFirstK(data,length,start,end,k);
}
// 找到数组中最后一个k的下标。如果数组中不存在k，返回-1
int GetLastK(int *data,int length,  int start ,int end ,int k)
{
	if(start > end)
		return -1;
	int middle = (start+end)/2;
	if(data[middle]==k)
	{
		if((middle<length-1 && data[middle+1]!=k) || middle==length-1)
			return middle;
		else 
			start = middle+1;
	}
	else if(data[middle]>k)
		end = middle-1;
	else
		start = middle+1;

	return GetLastK(data,length,start,end,k);
}
int GetNumberOfK(int *data, int length ,int k)
{	
	if(data==NULL || length<=0)
		return 0;
	int first = GetFirstK(data,length,0,length-1,k);
	int last = GetLastK(data,length,0,length-1,k);

	//cout<<last;
	if(first>-1 && last>-1)
		return last - first +1;
	return 0;
		
}
void Test(char *testName , int *data, int length ,int k, int expected)
{
	cout<<testName<<"\t";
	
	if(GetNumberOfK(data,length,k)==expected)
		cout<<"Passed.\n";
	else
		cout<<"Failed.\n";
	
}

// 查找的数字出现在数组的中间
void Test1()
{
    int data[] = {1, 2, 3, 3, 3, 3, 4, 5};
    Test("Test1", data, sizeof(data) / sizeof(int), 3, 4);
}

// 查找的数组出现在数组的开头
void Test2()
{
    int data[] = {3, 3, 3, 3, 4, 5};
    Test("Test2", data, sizeof(data) / sizeof(int), 3, 4);
}

// 查找的数组出现在数组的结尾
void Test3()
{
    int data[] = {1, 2, 3, 3, 3, 3};
    Test("Test3", data, sizeof(data) / sizeof(int), 3, 4);
}

// 查找的数字不存在
void Test4()
{
    int data[] = {1, 3, 3, 3, 3, 4, 5};
    Test("Test4", data, sizeof(data) / sizeof(int), 2, 0);
}

// 查找的数字比第一个数字还小，不存在
void Test5()
{
    int data[] = {1, 3, 3, 3, 3, 4, 5};
    Test("Test5", data, sizeof(data) / sizeof(int), 0, 0);
}

// 查找的数字比最后一个数字还大，不存在
void Test6()
{
    int data[] = {1, 3, 3, 3, 3, 4, 5};
    Test("Test6", data, sizeof(data) / sizeof(int), 6, 0);
}

// 数组中的数字从头到尾都是查找的数字
void Test7()
{
    int data[] = {3, 3, 3, 3};
    Test("Test7", data, sizeof(data) / sizeof(int), 3, 4);
}

// 数组中的数字从头到尾只有一个重复的数字，不是查找的数字
void Test8()
{
    int data[] = {3, 3, 3, 3};
    Test("Test8", data, sizeof(data) / sizeof(int), 4, 0);
}

// 数组中只有一个数字，是查找的数字
void Test9()
{
    int data[] = {3};
    Test("Test9", data, sizeof(data) / sizeof(int), 3, 1);
}

// 数组中只有一个数字，不是查找的数字
void Test10()
{
    int data[] = {3};
    Test("Test10", data, sizeof(data) / sizeof(int), 4, 0);
}

// 鲁棒性测试，数组空指针
void Test11()
{
    Test("Test11", NULL, 0, 0, 0);
}

int main(int argc, char* argv[])
{
    Test1();
    Test2();
    Test3();
    Test4();
    Test5();
    Test6();
    Test7();
    Test8();
    Test9();
    Test10();
    Test11();

    return 0;
}