HDU 4217 Data Structure?

线段树的单点更新的应用。

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

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#define N 1000000
__int64 temp;
struct node
{
    __int64 l,r;
    __int64 sum;
}xtree[4*N];
void pushup(int rn)
{
    xtree[rn].sum=xtree[rn<<1].sum+xtree[rn<<1|1].sum;
}
void build(__int64 l,__int64 r,__int64 rn)
{
    xtree[rn].l=l;
    xtree[rn].r=r;
    if(l==r)
    {
        xtree[rn].sum=1;
        return ;
    }
    __int64 mid=(l+r)>>1;
    build(l,mid,rn<<1);
    build(mid+1,r,rn<<1|1);
    pushup(rn);
}
void update(__int64 val,__int64 l,__int64 r,int rn)
{
    if(l==r)
    {
        temp+=l;
        xtree[rn].sum=0;
        return ;
    }
    int mid=(l+r)>>1;
    if(val<=xtree[rn<<1].sum)
    {
        update(val,l,mid,rn<<1);
    }
    else
    {
        update(val-xtree[rn<<1].sum,mid+1,r,rn<<1|1);
    }
    pushup(rn);
}
int main()
{
    __int64 n,m,a,t,c=1;
    scanf("%I64d",&n);
    while(n--)
    {
        temp=0;
        scanf("%I64d",&m);
        build(1,m,1);
        scanf("%I64d",&t);
        while(t--)
        {
            scanf("%I64d",&a);
            update(a,1,m,1);
        }
        printf("Case %I64d: ",c);
        printf("%I64d\n",temp);
        c++;
    }
    return 0;
}