【ZOJ】3279 Ants

#include<cstdio>
#define MAXN 100010
int tree[MAXN<<2];
inline void PushUp(int rt)
{
    tree[rt]=tree[rt<<1]+tree[rt<<1|1];
}
void Build(int L,int R,int rt)
{
    if(L==R)
        scanf("%d",&tree[rt]);
    else
    {
        int mid=(L+R)>>1;
        Build(L,mid,rt<<1);
        Build(mid+1,R,rt<<1|1);
        PushUp(rt);
    }
}
void Update(int x,int val,int L,int R,int rt)
{
    if(L==R)
        tree[rt]=val;
    else
    {
        int mid=(L+R)>>1;
        if(mid>=x)
            Update(x,val,L,mid,rt<<1);
        else
            Update(x,val,mid+1,R,rt<<1|1);
        PushUp(rt);
    }
}
int Query(int x,int L,int R,int rt)
{
    if(L==R)
        return L;
    int mid=(L+R)>>1;
    if(x<=tree[rt<<1])
        return Query(x,L,mid,rt<<1);
    else
    {
        x-=tree[rt<<1];
        return Query(x,mid+1,R,rt<<1|1);
    }
}
int main()
{
    int n,q,x,y;
    char ch;
    while(~scanf("%d",&n))
    {
        Build(1,n,1);
        scanf("%d",&q);
        while(q--)
        {
            scanf(" %c%d",&ch,&x);
            if(ch=='p')
            {
                scanf("%d",&y);
                Update(x,y,1,n,1);
            }
            else
                printf("%d\n",Query(x,1,n,1));
        }
    }
    return 0;
}