HDU Problem 4907 Take schedule 【二分】

Task schedule

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)

Total Submission(s): 2788    Accepted Submission(s): 949

Problem 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 <cstdio>
#include <cstring>
#include <algorithm>
#define MAX_N 200003

bool ju[MAX_N];
int ar[MAX_N];
int main() {
    int t, n, m, p, T;
    scanf("%d", &T);
    while (T--) {
        int cnt = 1;
        scanf("%d%d", &n, &m);
        int maxn = 0;
        memset(ju, false, sizeof(ju));
        for (int i = 1; i <= n; i++) {
            scanf("%d", &p);
            maxn = std::max(p, maxn);
            ju[p] = true;
        }
        
        //将ar数组里存放空闲的时间，对这些时间进行二分
        for (int i = 1; i <= maxn + 1; i++) {
            if (!ju[i]) ar[cnt++] = i;
        }
        while(m--) {
            scanf("%d", &t);
            
            //如果这个时间大于等于最大的空闲时间，
            //肯定在规定时间内可以完成
            if (t > maxn) {
                printf("%d\n", t);
                continue;
            }
            int ans = 0;
            int lb = 0, ub = cnt;
            while (ub - lb > 1) {
                int mid = (lb + ub)/2;
                if (ar[mid] >= t) {
                    ub = mid;
                }
                else {
                    lb = mid;
                }
            }
            printf("%d\n", ar[ub]);
        }
    }
    return 0;
}