HDU 4217 Data Structure?(线段树版）

题目链接

线段树 不熟啊。。。

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#define N 262145

__int64 sum;

struct node

{

    int l;

    int r;

    __int64 v;

}p[4*N];

void push(int rt)

{

    p[rt].v = p[rt<<1].v+p[rt<<1|1].v;

}

void build(int l,int r,int rt)

{

    int m;

    p[rt].l = l;

    p[rt].r = r;

    if(l == r)

    {

        p[rt].v = 1;

        return ;

    }

    m = (l + r)>>1;

    build(l,m,rt<<1);

    build(m+1,r,rt<<1|1);

    push(rt);

}

void update(int pos,int l,int r,int rt)

{

    int m;

    if(l == r)

    {

        sum += l;

        p[rt].v = 0;

        return ;

    }

    m = (l +r)>>1;

    if(pos > p[rt<<1].v)

    {

        update(pos-p[rt<<1].v,m+1,r,rt<<1|1);

    }

    else

    {

        update(pos,l,m,rt<<1);

    }

    push(rt);

}

int main()

{

    int i,n,m,t,num = 0,s;

    scanf("%d",&t);

    while(t--)

    {

        memset(p,0,sizeof(p));

        num ++;

        sum = 0;

        scanf("%d",&n);

        build(1,n,1);

        scanf("%d",&m);

        for(i = 1;i <= m;i ++)

        {

            scanf("%d",&s);

            update(s,1,n,1);

        }

        printf("Case %d: %I64d\n",num,sum);

    }

    return 0;

}