hdu 4007 Task schedule

利用树的思想，将一段连续的数据段分到一个树中，数据段中最小的数作为根，（借鉴别人的思想）

#include<iostream>
#include<cstdio>
#include<algorithm>

using namespace std;

#define N 100001
int root[2*N],s[2*N];

int main()
{
	int t;
	cin>>t;
	int a[2*N];
	while(t--)
	{
		int n,m;
		scanf("%d%d",&n,&m);
		int i;
		for(i=0;i<n;i++)
			scanf("%d",&a[i]);
		sort(a,a+n);
		memset(root,0,sizeof(root));
		memset(s,0,sizeof(s));
		root[a[0]]=a[0];
		s[a[0]]=1;
		for(i=1;i<n;i++)  //合并同一颗数；
		{
			if(a[i]==a[i-1]+1) {
				root[a[i]]=root[a[i-1]];
				s[root[a[i]]]++;
			}
			else
			{
				root[a[i]]=a[i];
				s[a[i]]=1;
			}
		}
		int x;
		for(i=1;i<=m;i++)
		{
			scanf("%d",&x);
			if(root[x])
			{
				printf("%d\n",root[x]+s[root[x]]);
			}
			else
				printf("%d\n",x);
		}


	}
return 0;
}