hdu4217 Data Structure?(线段树)

对线段树还不清楚的请看这个入门点击打开链接

看了一天才看懂。写下来希望自己能记得更牢

#include <stdio.h>
#define N 262144
struct node 
{
	long long left,right,count;
}tree[N*3];
void buildtree(long long p,long long l,long long r)//创建树，p是树的叶的编号，l表示某叶的编号的左区间，r表示某叶的编号的右区间。
{
	tree[p].left=l;
	tree[p].right=r;
	tree[p].count=r-l+1;//count表示此区间有多少个数
	if(tree[p].left==tree[p].right) return ;//当最区间等于右区间时结束递归。
	long long mid=(l+r)/2;//某叶的编号的下个左儿子一定是（l+r）/2，右儿子(l+r)/2+1;
	buildtree(2*p,l,mid);
	buildtree(2*p+1,mid+1,r);
}
int treefind(long long p,long long k)
{
	tree[p].count--;//每递归一次，更新区间数的个数
	if(tree[p].left==tree[p].right)//找到第k个小的数，当找到后k一定等于1
	return tree[p].left;
	if(tree[2*p].count>=k)//就在这里仔细多看看  理解理解 。左儿子区间的个数大于k
	treefind(2*p,k);
	else
	treefind(2*p+1,k-tree[2*p].count);//k的大小要减去左儿子区间的个数
}
int main()
{
	long long t,n,m,sum,x,q=1;
	scanf("%lld",&t);
	while(t--)
	{
		scanf("%lld %lld",&n,&m);
		buildtree(1,1,n);
		sum=0;
		while(m--)
		{
			scanf("%lld",&x);
			sum+=treefind(1,x);
		}
		printf("Case %lld: %lld\n",q++,sum);
	}
	return 0;
}