对线段树还不清楚的请看这个入门点击打开链接
看了一天才看懂。写下来希望自己能记得更牢
#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;
}