九度OJ 1349：数字在排序数组中出现的次数 （排序、查找）

时间限制：1 秒

内存限制：32 兆

特殊判题：否

提交：2489

解决：742

题目描述：

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

输入：

每个测试案例包括两行：

第一行有1个整数n，表示数组的大小。1<=n <= 10^6。

第二行有n个整数，表示数组元素，每个元素均为int。

第三行有1个整数m，表示接下来有m次查询。1<=m<=10^3。

下面有m行，每行有一个整数k，表示要查询的数。

输出：

对应每个测试案例，有m行输出，每行1整数，表示数组中该数字出现的次数。

样例输入：

81 2 3 3 3 3 4 513

样例输出：

4

思路：

先排序，NlogN复杂度。

对每个要查询的数，先查找这个数，然后搜索其左右，得到该数出现的个数，NlogN复杂度。

但该方法的最坏复杂度是MNlogN，如果测试数据不好，时间上无法AC。改进方法：排序后遍历统计每个数的出现个数，另设与原数组同样大小的数组来存储，这样查找到的数就能直接找到出现个数，复杂度MlogN。

代码：

#include <stdio.h>#include <stdlib.h> #define N 1000000#define M 1000 int cmp ( const void *a , const void *b ){    return *(int *)a - *(int *)b;} int main(void){    int n, m, i;    int a[N], b[M];     while (scanf("%d", &n) != EOF)    {        for(i=0; i<n; i++)            scanf("%d", &a[i]);        scanf("%d", &m);        for(i=0; i<m; i++)            scanf("%d", &b[i]);         qsort(a, n, sizeof(int), cmp);         int *p, *ptmp;        int count;        for(i=0; i<m; i++)        {            p = (int *)bsearch(&b[i], a, n, sizeof(a[0]), cmp);            //printf("find index = %d, key = %d\n", p-a, p[0]);            if (p == NULL)                count = 0;            else                count = 1;            ptmp = p;            if (p != NULL)            {                p++;                while (p - a <= n)                {                    if (*p == b[i])                    {                        p++;                        count++;                    }                    else                        break;                }            }            //printf("%d\n", count);            p = ptmp;            if (p != NULL)            {                p--;                while (p - a >= 0)                {                    if (*p == b[i])                    {                        p--;                        count++;                    }                    else                        break;                }            }            printf("%d\n", count);        }    }     return 0;}/**************************************************************    Problem: 1349    User: liangrx06    Language: C    Result: Accepted    Time:950 ms    Memory:8656 kb****************************************************************/