【HDU】4907 - Task schedule（并查集）

点击打开题目

Task schedule

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others) Total Submission(s): 2730 Accepted Submission(s): 921

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

Source

BestCoder Round #3

f 数组表示当天往后第一个空闲的天数，查询的时候同时进行路径压缩，如果查询的次数多的话，比二分的速度还要快。

代码如下：

#include <cstdio>
#define MAX 200000
int f[MAX+11];
int find(int x)
{
	if (x != f[x])
		f[x] = find (f[x]);
	return f[x];
}
int main()
{
	int u;
	int n,k;
	scanf ("%d",&u);
	while (u--)
	{
		scanf ("%d %d",&n,&k);
		for (int i = 1 ; i <= MAX ; i++)
			f[i] = i;
		int t;
		for (int i = 1 ; i <= n ; i++)
		{
			scanf ("%d",&t);
			f[t] = t + 1;
		}
		for (int i = 1 ; i <= k ; i++)
		{
			scanf ("%d",&t);
			printf ("%d\n",find(t));
		}
	}
	return 0;
}