题目描述:
有一台机器,并且给你这台机器的工作表,工作表上有n个任务,机器在ti时间执行第i个任务,1秒即可完成1个任务。
有m个询问,每个询问有一个数字q,表示如果在q时间有一个工作表之外的任务请求,请计算何时这个任务才能被执行。
机器总是按照工作表执行,当机器空闲时立即执行工作表之外的任务请求。
输入描述:
输入的第一行包含一个整数T, 表示一共有T组测试数据。
对于每组测试数据:
第一行是两个数字n, m,表示工作表里面有n个任务, 有m个询问;
第二行是n个不同的数字t1, t2, t3…tn,表示机器在ti时间执行第i个任务。
接下来m行,每一行有一个数字q,表示在q时间有一个工作表之外的任务请求。
特别提醒:m个询问之间是无关的。
[Technical Specification]
T <= 50
1 <= n, m <= 10^5
1 <= ti <= 2*10^5, 1 <= i <= n
1 <= q <= 2 * 10^5
输出描述:
对于每一个询问,请计算并输出该任务何时才能被执行,每个询问输出一行。
输入:
1
5 5
1 2 3 5 6
1
2
3
4
5
输出:
4
4
4
4
7
题意:
需要输出的是大于等于给定数且没有出现过的最小的数。
题解:
反着来
我们可以从最大位置开始枚举,用一个 now 变量来表示大于等于当前位置且为空的坐标
然后更新now
代码:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 200000 + 5;
int a[maxn],b[maxn];
int main(){
int t,n,m,x;
scanf("%d",&t);
while(t--){
scanf("%d%d",&n,&m);
memset(a,0,sizeof(a));
for(int i = 1; i <= n; i ++){
scanf("%d",&x);
a[x] = 1;
}
int now = maxn;
for(int i = maxn; i > 0; i --){
if(!a[i]) now = i;
b[i] = now;
}
for(int i = 1; i <= m; i ++){
scanf("%d",&x);
printf("%d\n",b[x]);
}
}
return 0;
}