HDU 4907 Task schedule

對於詢問q 如果q不存在直接輸出q

否則輸出後面第一個不存在的數

從2*10^5到1遍歷一邊ac

#include<bits/stdc++.h>
using namespace std;
int a[200020],b[200020];
int main()
{
    int T,n,m,t,q;
    scanf("%d",&T);
    for(int kase=1;kase<=T;kase++)
    {
        memset(a,0,sizeof(a));
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&t);
            a[t]=1;
        }
        int k=t+1;
        for(int i=200000;i>=1;i--)
        {
            if(a[i]==0)
                k=i;
            b[i]=k;
        }
        for(int i=1;i<=m;i++)   
        {
            scanf("%d",&q);
            printf("%d\n",b[q]);
        }
    }
    return 0;
}