九度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
, 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
****************************************************************/