HDU 4907 Task schedule

Description

有一台机器，并且给你这台机器的工作表，工作表上有n个任务，机器在ti时间执行第i个任务，1秒即可完成1个任务。
有m个询问，每个询问有一个数字q，表示如果在q时间有一个工作表之外的任务请求，请计算何时这个任务才能被执行。
机器总是按照工作表执行，当机器空闲时立即执行工作表之外的任务请求。

 

Input

输入的第一行包含一个整数T， 表示一共有T组测试数据。

对于每组测试数据：
第一行是两个数字n, m，表示工作表里面有n个任务, 有m个询问；
第二行是n个不同的数字t1, t2, t3....tn，表示机器在ti时间执行第i个任务。
接下来m行，每一行有一个数字q，表示在q时间有一个工作表之外的任务请求。

特别提醒：m个询问之间是无关的。

[Technical Specification]
1. T <= 50
2. 1 <= n, m <= 10^5
3. 1 <= ti <= 2*10^5, 1 <= i <= n
4. 1 <= q <= 2*10^5

 

Output

对于每一个询问，请计算并输出该任务何时才能被执行，每个询问输出一行。

 

Sample Input

1
5 5
1 2 3 5 6
1
2
3
4
5 

 

Sample Output

4
4
4
4
7 

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <cmath>
#define LL long long
#define INF 0x3f3f3f3f

using namespace std;

int a[100010];
int b[100010*2];
bool vis[100010*2];

void Init()
{
    for(int i=1;i<=100010*2;i++)
        b[i] = i;
}
int main()
{
    std::ios::sync_with_stdio(false);
    int T;
    cin>>T;
    while(T--)
    {
        memset(vis,false,sizeof(vis));
        int n,m;
        cin>>n>>m;
        for(int i=1;i<=n;i++)
        {
            cin>>a[i];
            vis[a[i]] = true;
        }
        sort(a+1,a+n+1);
        for(int i=a[n]+1;i>0;i--)
        {
            b[i] = i;
            if(vis[i])
                b[i] = b[i+1];
        }
        int q;
        while(m--)
        {
            cin>>q;
            if(!vis[q])
                cout<<q<<endl;
            else
                cout<<b[q]<<endl;
        }
    }
    return 0;
}