集训final G树状数组+二分查找

需要记得开一个数组存储数据以节省查询时间。。输入输出printf,scanf是必须滴。（这一次是数据小）

#include<iostream>



using namespace std;



typedef long long int llint;

const llint MAXN=100003;

llint tree[MAXN];

llint value[MAXN];

llint lowbit(llint idx)

{

    return idx & (-idx);

}

void add(llint pos, llint inc,llint n)

{

    for (llint i = pos; i <= n; i += lowbit(i))

    {

        tree[i] += inc;

    }

    return ;

}

llint sum_of(llint pos) 

{

    llint sum = 0;

    for (llint i = pos; i > 0; i -= lowbit(i)) 

    {

        sum += tree[i];

    }

    return sum;

}

int main()

{

    llint n;

    while(cin>>n)

    {

        llint i;

        memset(tree,0,sizeof(tree));

        memset(value,0,sizeof(value));

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

        {

            scanf("%I64d",&value[i]);

            add(i,value[i],n);

        }

        llint m;

        scanf("%I64d",&m);

        while(m--)

        {

            char s[10];

            scanf("%s",s);

            if(s[0]=='C')

            {

                llint a,b;

                scanf("%I64d%I64d",&a,&b);

                add(a,b-value[a],n);

                value[a]=b;

            }else if(s[0]=='Q')

            {

                llint pos;

                llint l,h,m;

                l=1;

                h=n;

                //cout<<n<<endl;

                bool tag=1;

                scanf("%I64d",&pos);

                //printf("%I64d|%I64d|%I64d\n",l,m,h);

                while(h>l&&(h-l)!=1)

                {

                    m=(l+h)/2;

                   // cout<<l<<"|"<<h;

                    //printf("%I64d|%I64d|%I64d\n",l,m,h);

                    llint sm=sum_of(m);

                    if(sm==pos)

                    {

                        while(value[m]==0&&m!=0)

                        {

                            m--;

                        }

                        cout<<m<<endl;

                        tag=0;

                        break;

                    }else if(sm>pos)

                    {

                        h=m;

                    }else

                    {

                        l=m;

                    }

                }

                if(tag)

                {

                    if(sum_of(l)>=pos)

                    {

                        while(value[l]==0&&l!=0)

                        {

                            l--;

                        }

                        cout<<l<<endl;

                    }else

                    {

                        while(value[h]==0&&h!=0)

                        {

                            h--;

                        }

                        cout<<h<<endl;

                    }

                }

            }

            

        }

    }

    return 0;

}