hdu 2817 Data Structure? （线段树）

线段树模板题 线段树的题目。向上更新。 

题意：

有N个数[ 1,2,3,,,,,n ]，取K个数，每次去第i小的数,求所有取出数的和！

 思路;

建立一个线段树，树节点记录的是节点所表示的区间内剩余数的个数（初始化时叶子节点的值都为1，然后向上传递）。

取数时，先找到第i小的数的叶子，值修改为1，然后向上传递。

#include<stdio.h>

#define N 30000000
struct node
{
int x,y,sum;
}aa[N*3];


void CreatTree(int t,int x,int y)
{
aa[t].x=x;
aa[t].y=y;
if(x==y)
{
aa[t].sum=1;
return;
}
int mid=(x+y)/2;
int temp=2*t;
CreatTree(temp,x,mid);
CreatTree(temp+1,mid+1,y);
aa[t].sum=aa[temp].sum+aa[temp+1].sum;
}


int FindTree(int t,int x)
{
int ans;
if(aa[t].x==aa[t].y)
{
aa[t].sum=0;
return aa[t].x;
}
int temp=2*t;
if(x<=aa[temp].sum)
ans=FindTree(temp,x);
else
ans=FindTree(temp+1,x-aa[temp].sum);
aa[t].sum=aa[temp].sum+aa[temp+1].sum;
return ans;
}


int main()
{
int T,n,m,k,r;
__int64 ans;
scanf("%d",&T);
r=1;

while(T--)
{
scanf("%d%d",&n,&m);
CreatTree(1,1,n);
ans=0;
while(m--)
{
scanf("%d",&k);
ans+=FindTree(1,k);
}
printf("Case %d: ",r++);
printf("%I64d\n",ans);
}
return 0;
}

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4217