正式赛1004 打猎的故事

Problem Description

小峰和爷爷来到了森林打猎。
森林里从左往右依次有 n 颗树排成一条直线，小峰站在最左边那颗树的左边。每棵树上都有一只鸟，第i 只鸟距离地面的高度为 h_i 。
当小峰站在距离地面高为 H 处向右开一枪后，所有离地面高为 H 的鸟都会被击落，其他鸟不动，并且鸟被击落后这棵树上就没有鸟了。
现在你知道小峰每次射击的高度，他想让你帮他算一算他每次射击能击落多少只鸟呢？

Input

第一行一个整数 T (1≤T≤10)，表示测试数据组数。接下来包含 T组测试数据。

对于每组测试数据，第一行输入两个数 n,q (1≤n,q≤10⁵) ，表示树的个数和小峰射击的次数。

第二行输入 n 个数 h₁,h₂⋯,hn(1≤h_i≤10⁹) ，表示每颗树上鸟的离地高度。

最后一行输入 q 个数，分别表示小峰每次射击的高度 H (1≤H≤10⁹) 。

Output

对于每组测试数据，请输出 q 个数，表示每次射击后被击落的鸟的数量。

Sample Input

1
4 4
4 1 3 4
1 1 4 2

Sample Output

1
0
2
0

这道题从题目上来说很简单，但是用常规方法做会时间超限，所以要优化一下，避免时间超限。
常规代码

#include<iostream>
using namespace std;
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        int n,q,sj;
        cin>>n>>q;
        int gd[n];
        for(int i=0;i<n;i++)
        {
            cin>>gd[i];
        }
        for(int i=0;i<q;i++)
        {
            int xn=0;
            cin>>sj;
            for(int i=0;i<n;i++)
            {
                if(sj==gd[i])
                {
                    xn++;
                    gd[i]=0;
                }
            }
            cout<<xn<<endl;
        }
    }
}

优化代码

scanf与printf同cin与cout相比效率更高。
#include<cstdio>
#include<cstring>
using namespace std;

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n,q,a,b,c[10002];
        scanf("%d %d",&n,&q);
        for(int i=0;i<n;i++)
        {
            scanf("%d",&a);
            c[a]++;
        }
        for(int i=0;i<q;i++)
        {
            scanf("%d",&b);
            printf("%d\n",c[b]);
            c[b]=0;
        }
    }
    return 0;
}