hdu4217Data Structure?

http://acm.hdu.edu.cn/showproblem.php?pid=4217

开一个数组标记这个数是否被拿走 s[w]为区间和 当求第K小的数的时候 让他与s[w]的左右子树比较 看在哪个区间中 依次找下去 总会找到一个s[]与他相等 所对应的区间就是那个应该被拿走的值

#include <iostream>
#include<cstdio>
#include<string.h>
#define MAX 262144
using namespace std;
int to[4*MAX];
__int64 s[4*MAX],sum;
void push(int w)
{
    s[w] = s[2*w]+s[2*w+1];
}
void ctree(int l, int r, int w)
{
    if(l==r)
    {
        s[w]=1;
        return ;
    }
    int m = (l+r)/2;
    ctree(l,m,2*w);
    ctree(m+1,r,2*w+1);
    push(w);//更新和
}
void add(int p,int l,int r,int w)
{
    if(l==r)
    {
        s[w] = 0;
        sum+=l;
        return ;
    }
    int m = (l+r)/2;
    if(p<=s[2*w])//在左右区间的哪个
    add(p,l,m,2*w);
    else
    {
        p-=s[2*w];//右区间剪掉左区间的和
        add(p,m+1,r,2*w+1);
    }
    push(w);//更新
}
int main()
{
    int i,j,k,n,m,t,a,x = 0;
    scanf("%d",&t);
    while(t--)
    {
        ctree(1,MAX,1);
        x++;
        sum = 0;
        scanf("%d%d",&n,&k);
        while(k--)
        {
            scanf("%d",&a);
            add(a,1,MAX,1);
        }
        printf("Case %d: ",x);
        printf("%I64d\n",sum);
    }
    return 0;
}