hdu 4007 Task schedule

最新推荐文章于 2021-04-29 22:00:49 发布

转载最新推荐文章于 2021-04-29 22:00:49 发布 · 548 阅读

模拟专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一种使用树结构来优化数据分段的方法，通过将数据段整合成树形结构，使得数据管理和查找效率得到显著提升。具体实现包括初始化根节点、合并相同层级的节点，以及对特定节点进行查询操作。

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

#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;
}