九度 题目1349：数字在排序数组中出现的次数

题目来源：http://ac.jobdu.com/problem.php?pid=1349

时间限制：1 秒

内存限制：32 兆

特殊判题：否

提交：1451

解决：432

题目描述：

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

输入：

每个测试案例包括两行：

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

分析：

    由于数组是排好顺序的，因此可以利用二分的思想找出重复k的第一个位置，以及最后一个位置，由此可以计算出K出现的次数。

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;

const int MAXN = 1000010;

int GetkFirstPosition(int* arr, int k, int iStart, int iEnd)//找出重复k的第一个位置
{
    if(iStart > iEnd)
        return -1;
    int iMid = (iStart + iEnd)>>1;
    int iMidData = arr[iMid];
    if(iMidData == k)
    {
        if((iMid > 0 && arr[iMid-1] != k) || iMid == 0)
            return iMid;
        else
            iEnd = iMid-1;
    }
    else if(iMidData > k)
        iEnd = iMid-1;
    else
        iStart = iMid+1;
    return GetkFirstPosition(arr, k, iStart, iEnd);
}

int GetKEndPosition(int* arr, int n, int k, int iStart, int iEnd)//找出重复k的最后一个位置
{
    if(iStart > iEnd)
        return -1;
    int iMid = (iStart + iEnd) >> 1;
    int iMidData = arr[iMid];
    if(iMidData == k)
    {
        if(((iMid < n-1) && arr[iMid+1] != k) || iMid == n-1)
            return iMid;
        else
            iStart = iMid+1;
    }
    else if(iMidData < k)
        iStart = iMid + 1;
    else
        iEnd = iMid - 1;
    return GetKEndPosition(arr, n, k, iStart, iEnd);
}

int GetNumOfK(int* arr, int n, int k)//计算k出现次数
{
    int iNum = 0;
    int iStart, iEnd;
    iStart = iEnd = 0;
    if(arr != NULL && n > 0)
    {
        iStart = GetkFirstPosition(arr, k, 0, n-1);
        iEnd = GetKEndPosition(arr, n, k, 0, n-1);
        if(iStart > -1 && iEnd > -1)
            iNum = iEnd - iStart + 1;
    }
    return iNum;
}

int main()
{
    int arr[MAXN], i, n, m, iData;
    while(~scanf("%d", &n))
    {
        for(i = 0; i < n; ++i)
            scanf("%d", &arr[i]);
        scanf("%d", &m);
        while(m--)
        {
            scanf("%d", &iData);
            int num = GetNumOfK(arr, n, iData);
            printf("%d\n", num);
        }
    }
    return 0;
}